pandas如何将相似的列名合并在一起？

import pandas as pd

# 创建一个示例数据框
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C_1': [7, 8, 9], 'C_2': [10, 11, 12]}
df = pd.DataFrame(data)

# 获取所有列名
columns = df.columns

# 使用字符串匹配找到相似的列名
similar_columns = [col for col in columns if 'C_' in col]

# 合并相似的列名
merged_column = '_'.join(similar_columns)

# 替换原始列名为新的合并列名
df.rename(columns={col: merged_column for col in similar_columns}, inplace=True)

# 打印结果
print(df)

   A  B  C
0  1  4  7
1  2  5  8
2  3  6  9