如何对列名进行分组并添加后缀？

import pandas as pd

# 示例数据
data = {'Column1': [1, 2, 3], 'Column2': [4, 5, 6], 'Column3': [7, 8, 9]}
df = pd.DataFrame(data)

# 分组依据
grouping_criteria = 'Column'

# 根据分组依据对列名进行分组并添加后缀
grouped_columns = {}
for column in df.columns:
    if grouping_criteria in column:
        group = column.split(grouping_criteria)[-1]  # 获取分组标识符
        if group not in grouped_columns:
            grouped_columns[group] = []
        grouped_columns[group].append(column)

# 对分组后的列名进行后缀添加
suffix = '_grouped'
for group, columns in grouped_columns.items():
    for column in columns:
        new_column = column + suffix
        df.rename(columns={column: new_column}, inplace=True)

# 打印分组后的列名
print(df.columns)