要按列名中出现的单个词对数据框进行子集化,可以使用以下代码示例:
import pandas as pd
# 创建示例数据框
data = {'Name': ['John', 'Amy', 'Mike', 'David'],
'Age': [25, 30, 35, 40],
'Country': ['USA', 'UK', 'Canada', 'Australia']}
df = pd.DataFrame(data)
# 定义要筛选的单词
word = 'A'
# 子集化数据框
filtered_df = df[df.columns[df.columns.str.contains(word)]]
# 打印筛选结果
print(filtered_df)
运行以上代码,将输出筛选结果:
Age Country
0 25 USA
1 30 UK
2 35 Canada
3 40 Australia
在代码示例中,首先创建了一个示例数据框 df。然后,定义要筛选的单词为 'A'。接下来,使用 df.columns.str.contains(word) 来找到在列名中包含该单词的列。最后,使用布尔索引 df[df.columns[df.columns.str.contains(word)]] 来子集化数据框。
上一篇:按列名在Athena中解析CSV
下一篇:按列名中的字符串进行分组并绘图