python - 我如何将字符串索引转换为整数索引

Question

数据框:


迎合
评分


一种
2.5

乙
4

C
3

一种
4

一种
2.5

乙
4


我想要的是:


迎合
评分
平均数


一种
2.5
3.25

乙
4
3.5

C
3
3

一种
4
3.25

乙
2.5
3.5

乙
4
3.5


我如何将服务索引转换为整数索引？

Answer 1

您需要使用自己的分组版本自行加入您的数据框。这可以通过使用 pd.merge() 来实现将您的数据框作为左表变量，将 groupby 数据框作为右表变量。

>>> df
   cater  rating
0      1     2.5
1      2     4.0
2      3     3.0
3      1     4.0
4      2     2.5
5      2     4.0
>>> df2 = df.groupby('cater', as_index=False).mean()
>>> df2
   cater  rating
0      1    3.25
1      2    3.50
2      3    3.00
>>> df3 = pd.merge(df, df2, on='cater', suffixes=('', '_average'))
>>> df3.assign(cater = df3['cater'].map({1:'a',2:'b',3:'c'}))
  cater  rating  rating_average
0     a     2.5            3.25
1     a     4.0            3.25
2     b     4.0            3.50
3     b     2.5            3.50
4     b     4.0            3.50
5     c     3.0            3.00