使用panda根據比較列中的值比較列值並創建列？

Question

我對python和熊貓很新。我有以下示例CSV作爲輸入。

ID Name Activity 
1 AB Play 
2 AD Hurt 
3 cd Polite 
4 CA Play 

我想這所以基本上轉化爲

ID Name Activity NumActivity 
1 AB Play  1 
2 AD Hurt  2 
3 cd Polite 3 
4 CA Play  1 

，我想給數值在爲活動列中的字符串。

因爲我對熊貓很新，所以我正在接近這個論壇尋求幫助。

感謝，

Answer 1

您可以使用factorize：

In [6]: 
df['NumActivity'] = pd.factorize(df['Activity'])[0] +1 
df 

Out[6]: 
    ID Name Activity NumActivity 
0 1 AB  Play   1 
1 2 AD  Hurt   2 
2 3 cd Polite   3 
3 4 CA  Play   1 

這裏factorize返回一個包含數組對一個元組：

In [8]: 
pd.factorize(df['Activity']) 

Out[8]: 
(array([0, 1, 2, 0]), Index(['Play', 'Hurt', 'Polite'], dtype='object')) 

所以我們的索引使用[0]，看到你的第一個元素希望您的索引從1開始，我們將+1設爲值。