📅  最后修改于: 2023-12-03 14:45:03.377000             🧑  作者: Mango
在 Pandas 中,使用列过滤器是非常常见的操作。通过列过滤器,我们可以选择需要的列,产生一个新的 DataFrame。
列过滤器可以使用 DataFrame 中的列名或者列数据的序号来进行选择。
我们可以使用方括号 []
来根据列名进行选择。
import pandas as pd
df = pd.read_csv('data.csv')
# 选择 'Name' 和 'Age' 列
df_new = df[['Name', 'Age']]
上面代码中,df[['Name', 'Age']]
中的双方括号表明选择了两个列名作为过滤器。
除了根据列名进行选择,我们还可以使用 iloc
方法根据列数据的序号进行选择。在 Pandas 中,行数据的序号从 0 开始。
import pandas as pd
df = pd.read_csv('data.csv')
# 选择第 0 和第 1 列,即 'Name' 和 'Age' 列
df_new = df.iloc[:, [0, 1]]
上面代码中,df.iloc[:, [0, 1]]
中的 :
表示选择所有行,[0, 1]
则表示选择第 0 和第 1 列。
使用列过滤器可以轻松地选择需要的列,并且可以方便地生成新的 DataFrame。