选择某几列是pandas中的基本操作之一。下面是详细讲解“pandas选择某几列的方法”的完整攻略,包括两个示例说明。
选择某几列的方法
通过列名选择
pandas中选择某几列最简单的方法是通过列名进行选择。可以使用[ ]
中传递的列名列表来选择需要的列。
示例:
import pandas as pd
# 创建一个数据表
data = {'country': ['China', 'Japan', 'USA', 'India'],
'population': [1395, 126.5, 329.5, 1353],
'area': [9.6, 0.4, 9.8, 3.3]}
df = pd.DataFrame(data)
# 选取 country 和 area 列
df2 = df[['country','area']]
print(df2)
输出结果为:
country area
0 China 9.6
1 Japan 0.4
2 USA 9.8
3 India 3.3
通过列的位置选择
在某些情况下,我们可能不知道列名,但知道其位置。在这种情况下,我们可以使用iloc
属性来选择列。
示例:
# 选取第一列和第三列
df3 = df.iloc[:,[0,2]]
print(df3)
输出结果为:
country area
0 China 9.6
1 Japan 0.4
2 USA 9.8
3 India 3.3
上述代码中的df.iloc[:,[0,2]]
表示选择所有行和第一列和第三列。
此外,还可以像选择某一列一样选择某一列的多个位置,例如:
# 选取第一列和第三列,用逗号表示多个位置
df4 = df.iloc[:, [0,2]]
print(df4)
输出结果为:
country area
0 China 9.6
1 Japan 0.4
2 USA 9.8
3 India 3.3
总结
本文介绍了pandas选择某几列的方法,包括通过列名选择和通过列的位置选择。使用这些方法可以轻松地选择需要的列。