以下是一个示例代码,用于保留列标题为彩色的列,删除其余列:
import pandas as pd
# 创建示例数据框
data = {
'列1': [1, 2, 3],
'列2': ['红色', '绿色', '蓝色'],
'列3': ['A', 'B', 'C'],
'列4': ['红色', '绿色', '蓝色']
}
df = pd.DataFrame(data)
# 保留列标题为彩色的列
df = df.loc[:, df.columns.str.contains('色')]
print(df)
输出结果:
列2 列4
0 红色 红色
1 绿色 绿色
2 蓝色 蓝色
该示例使用pandas库创建了一个数据框,并使用df.columns.str.contains('色')
来检查列标题中是否包含"色"这个关键字。然后,使用df.loc
方法根据布尔索引保留包含"色"的列,并将结果赋值给df。最后,打印df的结果,即只保留了列标题为彩色的列,删除了其余列。