我有一个像这样定义的列列表:
我有一个数据框df. 如果任何一列与colname匹配,那么我希望从这个列名中删除_the_x。
我应该如何做?
col_list=['Name_x','Num_x']
我有一个数据框df. 如果任何一列与colname匹配,那么我希望从这个列名中删除_the_x。
我应该如何做?
# df.columns
# Index(['Name_x', 'Num_x', 'test_x'], dtype='object')
col_list=['Name_x','Num_x']
df.columns = np.where(
df.columns.isin(col_list), df.columns.str.replace(r'_x$', ''), df.columns)
# df.columns
# Index(['Name', 'Num', 'test_x'], dtype='object')