Python 如何在pandas 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”对象没有属性“add_categories”

如果列名称中没有catogorical，则仅使用

平均值

：

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

或者，如果要将列名转换为分类索引，请使用：

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

---

如果列名称中没有CATOGORICAL，则仅使用

mean

：

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

或者，如果要将列名转换为分类索引，请使用：

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