python - 如何根据列名删除列python pandas

Question

我想删除数据框中以“y”结尾的每一列。出于某种原因，我拥有的数据将每列列出两次，唯一不同的是列名，如下所示:

d = {'Team': ['1', '2', '3'], 'Team_y': ['1', '2', '3'], 'Color' : ['red', 'green', 'blue'], 'Color_y' : ['red', 'green', 'blue']}
df = pd.DataFrame(data=d)
df

    Team    Team_y  Color   Color_y
0    1        1      red     red
1    2        2     green   green
2    3        3      blue    blue

我知道这是某种字符串格式。我尝试使用 [-1] 索引最后一个字母，但无法让它正常工作。谢谢!

Answer 1

drop column based on a string condition

df.drop([col for col in df.columns if '_y' in col],axis=1,inplace=True)

更好的是，如果它必须特定于以它结尾，那么:

df.drop([col for col in df.columns if col.endswith('_y')],axis=1,inplace=True)