重命名dataframe中具有与旧名称相似的字符串的列名

import pandas as pd

def rename_similar_columns(df, old_name, new_name):
    new_columns = {}
    for column in df.columns:
        if old_name in column:
            new_column = column.replace(old_name, new_name)
            new_columns[column] = new_column
    df.rename(columns=new_columns, inplace=True)

# 示例用法
df = pd.DataFrame({'old_name_1': [1, 2, 3], 'old_name_2': [4, 5, 6], 'other_column': [7, 8, 9]})
print("原始dataframe:")
print(df)

rename_similar_columns(df, 'old_name', 'new_name')
print("重命名后的dataframe:")
print(df)

原始dataframe:
   old_name_1  old_name_2  other_column
0           1           4             7
1           2           5             8
2           3           6             9
重命名后的dataframe:
   new_name_1  new_name_2  other_column
0           1           4             7
1           2           5             8
2           3           6             9