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.Categorical ：

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

Answer 2

您可以使用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 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