Pandas：具有另一列唯一值索引的新列

Question

我的数据帧：

ID       Name_Identify  ColumnA  ColumnB  ColumnC
1        POM-OPP        D43      D03      D59
2        MIAN-ERP       D80      D74      E34
3        POM-OPP        E97      B56      A01
4        POM-OPP        A66      D04      C34
5        DONP28         B55      A42      A80
6        MIAN-ERP       E97      D59      C34

预期的新数据帧：

ID       Name_Identify ColumnA  ColumnB  ColumnC    NEW_ID
1        POM-OPP       D43      D03      D59        1
2        MIAN-ERP      D80      D74      E34        2
3        POM-OPP       E97      B56      A01        1
4        POM-OPP       A66      D04      C34        1
5        DONP28        B55      A42      A80        3
6        MIAN-ERP      E97      D59      C34        2

Answer 1

您可以使用pandas.groupby：

df['NEW_ID'] = df.groupby('Name_Identify', sort=False).ngroup() + 1
print(df)

印刷品：

   ID Name_Identify ColumnA ColumnB ColumnC  NEW_ID
0   1       POM-OPP     D43     D03     D59       1
1   2      MIAN-ERP     D80     D74     E34       2
2   3       POM-OPP     E97     B56     A01       1
3   4       POM-OPP     A66     D04     C34       1
4   5        DONP28     B55     A42     A80       3
5   6      MIAN-ERP     E97     D59     C34       2

Answer 2

您可以使用pandas.Categorical：

df["NEW_ID"] = pd.Categorical(df["Name_Identify"], ordered=False).codes + 1

Answer 3

convert = {k: v for v, k in enumerate(df.Name_Identify.unique(), start=1)}
df["NEW_ID"] = df.Name_Identify.map(convert)

---

解释：

在第一个命令中，我们从Name_Identify列中选择唯一名称

In[23]: df.Name_Identify.unique()

array(['POM-OPP', 'MIAN-ERP', 'DONP28'], dtype=object)

然后从它们的枚举序列创建一个字典（枚举以1开始）：

In[24]: convert = {k: v for v, k in enumerate(df.Name_Identify.unique(), start=1)}
In[25]: convert

{'POM-OPP': 1, 'MIAN-ERP': 2, 'DONP28': 3}

在第二个命令中，我们使用这个字典通过将Name_Identify列中的所有名称转换为适当的数字来创建一个新列：

In[26]: df["NEW_ID"] = df.Name_Identify.map(convert)
In[27]: df

   D Name_Identify ColumnA ColumnB ColumnC  NEW_ID
0  1       POM-OPP     D43     D03     D59       1
1  2      MIAN-ERP     D80     D74     E34       2
2  3       POM-OPP     E97     B56     A01       1
3  4       POM-OPP     A66     D04     C34       1
4  5        DONP28     B55     A42     A80       3
5  6      MIAN-ERP     E97     D59     C34       2