Python DataFrame：如何保留特定关键字的列

假设有一个数据框，名为 df，其中包含多列，您只想保留包含特定关键字的列，可以使用以下代码：

# 保留包含特定关键字的列
keyword = 'key' # 指定关键字
df = df.loc[:, df.columns.str.contains(keyword)]

这里使用了 .loc 方法来选取列，而且对列名进行了筛选，只保留包含关键字的列。这里的 str.contains() 方法用于判断列名是否包含指定的关键字。