在Python中,可以使用pandas库的DataFrame对象来进行按列过滤数据框。以下是一个示例代码:
import pandas as pd
# 创建一个包含多列的数据框
data = {'A': [1, 2, 3, 4, 5],
'B': [10, 20, 30, 40, 50],
'C': [100, 200, 300, 400, 500]}
df = pd.DataFrame(data)
# 按列过滤数据框,只保留指定的列
filtered_df = df[['A', 'C']]
print(filtered_df)
以上代码首先创建了一个包含多列的数据字典,然后使用pandas的DataFrame函数将数据字典转换为数据框。接下来,使用df[['A', 'C']]
的语法来指定只保留列'A'和'C',并将结果赋给一个新的数据框变量filtered_df
。最后,使用print(filtered_df)
来打印过滤后的数据框。
输出结果为:
A C
0 1 100
1 2 200
2 3 300
3 4 400
4 5 500
这样就实现了按列过滤数据框的功能。你可以根据自己的需求更改列名或添加更多的列。
下一篇:按列过滤数据框中的最大和最小值