从数据框 Python Pandas 中选择 2 列

在Python Pandas中选择数据框的列是非常简单的。可以使用以下代码从数据框中选择两列：

import pandas as pd

# 创建数据框
data = {'Name': ['Tom', 'John', 'Smith', 'Jack', 'David'],
        'Age': [25, 27, 22, 29, 24],
        'Country': ['USA', 'UK', 'Canada', 'Australia', 'India']}
df = pd.DataFrame(data)

# 选择两列
df_two_columns = df[['Name', 'Age']]

这将会创建一个数据框 df，它包含 Name, Age, 和 Country 这三个列。接着使用双方括号运算符来选择 Name 和 Age 两列，并将它们保存到 df_two_columns 中。

如果想要选择更多或更少的列，只需修改双方括号中的列名即可。

此外，还可以使用 .loc 和 .iloc 来选择行和列的子集，以及使用 .drop 来从数据框中删除一列。

• 使用 loc 选择行和列

# 选择第1到3行和第1到2列
df_sub = df.loc[1:3, ['Name', 'Age']]

# 选择所有行和第2列
df_sub2 = df.loc[:, ['Age']]

• 使用 iloc 选择行和列

# 选择第1到3行和第1到2列
df_sub = df.iloc[1:4, [0, 1]]

# 选择所有行和第2列
df_sub2 = df.iloc[:, [1]]

• 使用 drop 删除列

# 删除列 'Country'
df_drop = df.drop(['Country'], axis=1)

以上为选择数据框中两列的三种方法，不同的方法可以根据不同的需求来选择。