python - 在 agg 函数中聚合具有一个属性的多列
问题描述
假设我有一个data_stores
类似于以下内容的 pandas dataFrame ():
store| item1 | item2 | item3
------------------------------
1 | 45 | 50 | 53
1 | 200 | 300 | 250
2 | 20 | 17 | 21
2 | 300 | 350 | 400
假设我想在列item1
上聚合mean
item2
item3
sum
.
这通常可以通过以下方式完成:
data_stores_total= data_stores.groupby(['store'], as_index=False).agg({'item1': 'mean', 'item2': 'sum', 'item3': 'sum' })
但是,这不能通过以下方式(更有效地)完成:
data_stores_total= data_stores.groupby(['store'], as_index=False).agg({'item1': 'mean', ['item2', 'item3']: 'sum' })
以下方式对字典键更有意义:
data_stores_total= data_stores.groupby(['store'], as_index=False).agg({'mean': 'item1':, 'sum': ['item2', 'item3']})
有没有办法在数据框的某些列上使用相同的函数进行聚合,而无需在agg
每个列的函数中编写新的字典属性?
解决方案
这是不可能的,只有你可以用函数的键和列名的列表定义字典,然后在循环中用值交换键:
data_stores = pd.DataFrame({'store': [1, 1, 2, 2],
'item1': [45, 200, 20, 300],
'item2': [50, 300, 17, 350],
'item3': [53, 250, 21, 400]})
print (data_stores)
store item1 item2 item3
0 1 45 50 53
1 1 200 300 250
2 2 20 17 21
3 2 300 350 400
d = {'mean':'item1', 'sum' : ['item2', 'item3']}
out = {}
for k, v in d.items():
if isinstance(v, list):
for x in v:
out[x] = k
else:
out[v] = k
print (out)
{'item1': 'mean', 'item2': 'sum', 'item3': 'sum'}
data_stores_total = data_stores.groupby('store', as_index=False).agg(out)
print (data_stores_total)
store item1 item2 item3
0 1 122.5 350 303
1 2 160.0 367 421
或者:
d = {'mean':['item1'], 'sum' : ['item2', 'item3']}
d1 = {k: oldk for oldk, oldv in d.items() for k in oldv}
print (d1)
{'item1': 'mean', 'item2': 'sum', 'item3': 'sum'}
data_stores_total = data_stores.groupby('store', as_index=False).agg(d1)
print (data_stores_total)
store item1 item2 item3
0 1 122.5 350 303
1 2 160.0 367 421
编辑:
如果想通过相同的聚合函数聚合所有没有很少的列,您可以按所有列创建字典,并按列表过滤,difference
然后添加缺失的对键:列的值:聚合函数:
out = dict.fromkeys(data_stores.columns.difference(['store','item1']), 'sum')
out['item1'] = 'mean'
print (out)
{'item2': 'sum', 'item3': 'sum', 'item1': 'mean'}
data_stores_total = data_stores.groupby('store', as_index=False).agg(out)
print (data_stores_total)
store item2 item3 item1
0 1 350 303 122.5
1 2 367 421 160.0
您还可以传递使用此列的自定义函数:
def func(x):
return x.sum() / x.mean()
out = dict.fromkeys(data_stores.columns.difference(['store','item1']), 'sum')
out['item1'] = func
print (out)
{'item2': 'sum', 'item3': 'sum', 'item1': <function func at 0x000000000F3950D0>}
data_stores_total = data_stores.groupby('store', as_index=False).agg(out)
print (data_stores_total)
store item2 item3 item1
0 1 350 303 2
1 2 367 421 2
推荐阅读
- python - 在 Python 中结合 Perlin 噪声和泊松圆盘采样
- python - 如何在pygame中检测Sprite和Rect之间的碰撞
- javascript - React Native:以多种颜色动态打印文本
- javascript - 如何使用 PHP 从 SQL Server 填充 HTML 表单字段
- javascript - 制作变量时未定义$
- html - 大小不等的 div 无法相对于顶部居中
- arrays - How to print a circulant matrix (toeplitz), in MATLAB, where each input is a 3D matrix of dimention m x m x 3?
- javascript - 如何分离和显示第二选择数据?
- android - 收到图片时应用程序崩溃
- docker - 未找到 Docker 容器 NPM 依赖项