如何按熊貓數據框循環分組？

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)