如何基於字符串列

Question

我有一個熊貓數據幀像這樣的gouping的熊貓數據幀創建新列：

df = pd.DataFrame({'a':['A','A','A','B','B'], 
        'b':['Alabama','Alabama','Antioch','Brisbane','Boolean']}) 

我要添加兩個新列說，「N1」和「N2」等等屬於第一組'a'列的所有行將具有值1，並且下一組將在新列'n1'中具有值2。相同的邏輯列「b」和「N 2」，但嵌套的「A」

所得數據幀分組內應該是這樣的（由於某種原因，降價表不因此工作中的鏈接）：

我該怎麼做？

Answer 1

您可以使用pandas.factorize至將輸入值編碼爲枚舉類型或分類變量;要通過a得到n2列，只是組和因式分解b：

import pandas as pd 

df['n1'] = pd.factorize(df.a)[0] + 1 
df['n2'] = df.groupby('a').b.transform(lambda x: pd.factorize(x)[0] + 1) 

df 
# a   b n1 n2 
#0 A Alabama 1 1 
#1 A Alabama 1 1 
#2 A Antioch 1 2 
#3 B Brisbane 2 1 
#4 B Boolean 2 2 

Answer 2

或者使用astypecategory

df['n1'] = df.a.astype('category').cat.codes.add(1) 
df['n2'] = df.groupby('a').b.transform(lambda x: x.astype('category').cat.codes.add(1)) 

df 
Out[1254]: 
    a   b n1 n2 
0 A Alabama 1 1 
1 A Alabama 1 1 
2 A Antioch 1 2 
3 B Brisbane 2 2 
4 B Boolean 2 1