Pandas如何使用聚合计算列表x中列的总和和列表y中列的平均值

vyswwuz2 于 2022-11-05 发布在其他

关注(0)|答案(1)|浏览(142)

我有一个包含60多个列的 Dataframe 。我想计算以'FLAG_'开头的列名的sum和以'RESPONSE_'开头的列名的mean。如何实现这一点？我尝试使用agg（），但没有成功。

注意：我正在尝试将此作为数据管道得一部分

sum_cols = [col for col in df.columns if 'FLAG_' in col]
mean_cols = [col for col in df.columns if 'RESPONSE_' in col]
dict_ = {tuple(sum_cols): 'sum', tuple(mean_cols): 'mean'}
df = df.groupby('ID').agg(dict_)

但是我得到了错误，因为在选择列时，它应该是一个列表而不是元组。因为它是一个字典，我不能将列表作为键传递。

KeyError: "Column(s) [array([#<All the FLAG_ columns listed>], dtype=object)] do not exist"

pandas

来源：https://stackoverflow.com/questions/74284604/pandas-how-to-calculate-sum-for-columns-in-list-x-and-mean-for-columns-in-list-y

1条答案

按热度按时间

wj8zmpe11#

sum_cols = [col for col in df.columns if 'FLAG_' in col]
    mean_cols = [col for col in df.columns if 'RESPONSE_' in col]
    dict_ = {col:'sum' for col in sum_cols}
    dict_.update({col:'mean' for col in mean_cols})
    df = df.groupby('ID').agg(dict_)

赞(0）回复(0）举报 2022-11-05

我来回答

Pandas如何使用聚合计算列表x中列的总和和列表y中列的平均值

1条答案

相关问题

热门标签

最新问答