Python：如何将数字映射到列中的唯一项目（枚举唯一对象）？

Question

我需要将数字分配给数据框中特定列的唯一值，并将其作为新列...例如：ex_list=['12-B', '10002', '3A', '4', 'DCX', '3A'] df=pd.DataFrame(ex_list, columns=['Items'])这是预期的输出：请注意，我需要对超过 1M 行的数据框执行此操作，因此肯定会喜欢一种有效的方法！

Answer 1

你可以这样做。

import pandas as pd

ex_list=['12-B', '10002', '3A', '4', 'DCX', '3A']

df=pd.DataFrame(ex_list, columns=['Items'])

df['Num_Items']=pd.factorize(df['Items'].tolist())[0]

输出

    Items   Num_Items

0   12-B    0

1   10002   1

2   3A      2

3   4       3

4   DCX     4

5   3A      2

Answer 2

ex_list=['12-B', '10002', '3A', '4', 'DCX', '3A']

df=pd.DataFrame(ex_list, columns=['Items'])

从列中获取唯一值，在这些唯一项和整数之间创建映射，然后沿Items列应用该映射。

mapping = {item:i for i, item in enumerate(df["Items"].unique())}

df["Num_Items"] = df["Items"].apply(lambda x: mapping[x])

df