python - 如何将聚合函数应用于 Pandas 中数据透视表的所有列

Question

数据透视表正在计算一种现象的每月发生次数。这是简化的示例数据，后面是枢轴：

+--------+------------+------------+
| ad_id  | entreprise | date       |
+--------+------------+------------+
| 172788 | A          | 2020-01-28 |
| 172931 | A          | 2020-01-26 |
| 172793 | B          | 2020-01-26 |
| 172768 | C          | 2020-01-19 |
| 173219 | C          | 2020-01-14 |
| 173213 | D          | 2020-01-13 |
+--------+------------+------------+

我的 pivot_table 代码如下：

my_pivot_table = pd.pivot_table(df[(df['date'] >= some_date) & ['date'] <= some_other_date)], 
                                values=['ad_id'], index=['entreprise'], 
                                columns=['year', 'month'], aggfunc=['count'])

结果表如下所示：

+-------------+---------+----------+-----+----------+
|             |  2018   |          |     |          |
+-------------+---------+----------+-----+----------+
| entreprise  | january | february | ... | december |
| A           | 12      | 10       | ... | 8        |
| B           | 24      | 12       | ... | 3        |
| ...         | ...     | ...      | ... | ...      |
| D           | 31      | 18       | ... | 24       |
+-------------+---------+----------+-----+----------+

现在，我想添加一个列，为我提供月平均值，并执行其他操作，例如将上个月的计数与过去 12 个月的月平均值进行比较......

我试图摆弄 pivot_table 的 aggfunc 参数，并尝试将平均列添加到原始数据框中，但没有成功。

提前致谢！

标签： pythonpandaspivot-tableaggregate-functions

Answer 1

因为您可以使用Multiindex后获得表格pivot_table：

df1 = df.mean(axis=1, level=0)
df1.columns = pd.MultiIndex.from_product([df1.columns, ['mean']])

或者：

df2 = df.mean(axis=1, level=1)
df2.columns = pd.MultiIndex.from_product([['all_years'], df2.columns])