如何在python中根据条件重命名列名？

import pandas as pd

data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)

condition = df.columns == 'A'  # 选择列名为'A'的列

df.columns = ['X' if condition else col for col, condition in zip(df.columns, condition)]

import pandas as pd

data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)

condition = df.columns == 'A'
df.columns = ['X' if condition else col for col, condition in zip(df.columns, condition)]

print(df)

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