Python Groupby часть строки

Question

Я группирую список транзакций по британскому почтовому индексу, но я хочу группировать только по первой части почтового индекса. Таким образом, британские почтовые коды находятся в двух частях, наружу и внутрь, разделенных пробелом. например W1 5DA.

subtotals = df.groupby('Postcode').count() 

ли способ, что я делаю это сейчас, как я думал, что делать это на данный момент, является добавлением столбца в DataFrame только с первым словом столбца Почтового индекса, а затем группировкой по что ... но мне интересно, есть ли более простой способ сделать это.

Спасибо

Answer 1

Я думаю, что вам нужно groupby по Series созданный split первым пространством:

subtotals = df.groupby(df['Postcode'].str.split().str[0]).count() 

Пример:

df = pd.DataFrame({'Postcode' :['W1 5DA','W1 5DA','W2 5DA']}) 
print (df) 
    Postcode 
0 W1 5DA 
1 W1 5DA 
2 W2 5DA 

print (df['Postcode'].str.split().str[0]) 
0 W1 
1 W1 
2 W2 
Name: Postcode, dtype: object 

subtotals = df.groupby(df['Postcode'].str.split().str[0]).count() 
print (subtotals) 
      Postcode 
Postcode   
W1    2 
W2    1 

Проверить также What is the difference between size and count in pandas?