Python dataframe以3列为一组添加列

import pandas as pd

# 创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
                   'B': [6, 7, 8, 9, 10],
                   'C': [11, 12, 13, 14, 15],
                   'D': [16, 17, 18, 19, 20],
                   'E': [21, 22, 23, 24, 25],
                   'F': [26, 27, 28, 29, 30]})

# 将DataFrame的列按每三列一组进行分割
groups = [df.columns[i:i+3] for i in range(0, len(df.columns), 3)]

# 遍历每一组列，并在每组后面添加一列
for group in groups:
    group_name = '_'.join(group)  # 组名为列名的连接
    df[group_name] = df[group].sum(axis=1)  # 在每组后面添加一列，该列为该组列的和

# 打印添加列后的DataFrame
print(df)

   A   B   C   D   E   F  A_B_C  D_E_F
0  1   6  11  16  21  26     18     63
1  2   7  12  17  22  27     21     66
2  3   8  13  18  23  28     24     69
3  4   9  14  19  24  29     27     72
4  5  10  15  20  25  30     30     75