Use this snippet in order to replace a string in column names for a pandas DataFrame:
new_df = df.rename(columns=lambda s: s.replace("A", "B")) # df will not be modified !You can also modify the column names in-place (i.e. modify the original DataFrame):
df.rename(columns=lambda s: s.replace("A", "B"), inplace=True)For example, if you have the columns ["ColumnA", "X", "Y"] before running .rename(), the result will have ["ColumnB", "X", "Y"] (the "A" has been replaced by "B")