如何通过 pandas 数据帧循环分组？

Question

我是 Python 初学者，我有以下代码：levage_mean = df.groupby('cum_cycle_id')['levage'].mean().rename("levage_mean")levage_min = df.groupby('cum_cycle_id')['levage'].min().rename("levage_min")levage_max = df.groupby('cum_cycle_id')['levage'].max().rename("levage_max")levage_sum = df.groupby('cum_cycle_id')['levage'].sum().rename("levage_sum")load_min= df.groupby('cum_cycle_id')['load'].min().rename("load_min")load_max = df.groupby('cum_cycle_id')['load'].max().rename("load_max")load_mean = df.groupby('cum_cycle_id')['load'].mean().rename("load_mean")load_sum = df.groupby('cum_cycle_id')['load'].sum().rename("load_sum")load_median = df.groupby('cum_cycle_id')['load'].median().rename("load_median")`我想把它放在一个循环中，因为它太重复了，我该怎么办？谢谢您的回复！

Answer 1

你可以试试这个：

init_list = []

my_list = ['leaf_', 'load_']

mean_min_max_list = ['mean', 'min', 'max', 'sum']

for item in my_list:

    for ele in mean_min_max_list:

        curr = df.groupby('cum_cycle_id')[item].mean().rename(item + ele)

        init_list.append(curr)