pandas 选择某几列的方法

  • Post category:Python

选择某几列是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选择某几列的方法,包括通过列名选择和通过列的位置选择。使用这些方法可以轻松地选择需要的列。