如何在pandas groupby中添加新列？

Question

我有熊貓 groupby 對象

| A  | B | C |
|----|---|---|
| a1 | 3 | 4 |
| a2 | 6 | 2 |

我想添加新列，該列將具有行格式的平均值

所需輸出

| A  | B | C | Average |
|----|---|---|---------|
| a1 | 3 | 4 | 3.5     |
| a2 | 6 | 2 | 4       |

我試過

df.columns = df.columns.add_categories(['Average'])
df['Average'] = df.mean(axis = 1)

但這給了我

AttributeError: 'Index' object has no attribute 'add_categories'

Answer 1

如果列名稱中沒有分類，則僅使用mean ：

df['Average'] = df.mean(axis = 1)

或者，如果要將列名稱轉換為CategoricalIndex使用：

df.columns = pd.CategoricalIndex(df.columns)
df.columns = df.columns.add_categories(['Average'])
df['Average'] = df.mean(axis = 1)