Python Groupby 字符串的一部分

Question

我正在按英国邮政编码对交易列表进行分组，但我只想按邮政编码的第一部分进行分组。因此，英国邮政编码分为两部分，向外和向内，由 [空格] 分隔。例如W1 5DA。

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

我现在的做法是，我现在考虑的做法是在 DataFrame 中添加另一列，仅包含邮政编码列的第一个单词，然后按此分组...但我想知道是否有更简单的方法。

谢谢

Answer 1

我想你需要 groupby by Series created by 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?