Python 来自多组的新列

Question

我有一个简化的 dataframe，可以说：

df = pd.DataFrame({'Letter': ['A', 'B', 'A', 'B','A','B','A','B'], 'T/F': [True, True, False, False,True,False,True,False], 'Number':[5, 6, 7, 8, 9,10,11,12]})

我有一些其他的 df2 已经有（isAB 或 isTF）的列。 我想要做的是使我的新列类似于 df2.loc[:,'A_True] ，它具有 A 和 True 的数字平均值。 如果我执行以下操作：

df.groupby(['Letter','T/F'])['Number'].mean()[1::2]

这输出

Letter  T/F 
A       True    8.333333
B       True    6.000000
Name: Number, dtype: float64

我希望我的 df2 有一列就是 8.33333, 6。是否需要合并？ 我想节省时间而不是做

 for name,group in df.groupby('Letter'):
     df2.loc[df['Letter']==A,'A_True'] = group.loc[group['T/F'==True],'Number'].mean()

Answer 1

可以先筛选

df.loc[df['T/F']].groupby('Letter')['Number'].mean()
Out[93]: 
Letter
A    8.333333
B    6.000000
Name: Number, dtype: float64

Answer 2

我们可以使用DataFrame.unstack ：

new_df=df.groupby(['Letter','T/F'],sort=False).Number.mean().unstack()
print(new_df)

T/F        True   False
Letter                 
A       8.333333    7.0
B       6.000000   10.0

---

new_df[True]

Letter
A    8.333333
B    6.000000
Name: True, dtype: float64

---

编辑

new_df=df.groupby(['Letter','T/F'],sort=False).Number.mean().unstack().reset_index()
print(new_df)

T/F Letter      True  False
0        A  8.333333    7.0
1        B  6.000000   10.0