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

Question

我需要将数字分配给数据框中特定列的唯一值，并将其作为新列...

例如：

ex_list=['12-B', '10002', '3A', '4', 'DCX', '3A']
df=pd.DataFrame(ex_list, columns=['Items'])

这是预期的输出：

请注意，我需要对超过 1M 行的数据框执行此操作，因此肯定会喜欢一种有效的方法！

标签： pythonpandasdataframemappingenumerate

Answer 1

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