在 pandas 数据框中选择列 - Python

在使用 Pandas 数据框时，我们经常需要选择特定的列来进行操作。在这里，我们将介绍如何使用 Pandas 数据框中的列名，选择想要的列进行操作。

选择单列

要选择单列，请使用以下操作：

import pandas as pd

# 创建数据框
df = pd.DataFrame({'name': ['John', 'Mike', 'Sarah'], 'age': [25, 30, 28], 'city': ['New York', 'Los Angeles', 'Chicago']})

# 选择单列
df['name']

上面的代码输出：

0     John
1     Mike
2    Sarah
Name: name, dtype: object

在这里，我们使用了数据框名称 df 和要选择的列名称 name。结果是 df 数据框中 name 列的所有值。

选择多列

要选择多个列，请使用以下操作：

# 选择多列
df[['name', 'age']]

上面的代码输出：

    name  age
0   John   25
1   Mike   30
2  Sarah   28

在这里，我们使用了数据框名称 df 和要选择的列名称 name 和 age。结果是 df 数据框中 name 和 age 列的所有值。

选择连续的列

如果要选择连续的列，请使用以下操作：

# 选择连续的列
df.loc[:, 'name':'age']

上面的代码输出：

    name  age
0   John   25
1   Mike   30
2  Sarah   28

在这里，我们使用了 loc 函数选择了从列名 name 到列名 age 之间的所有列。结果与选择多列相同。

总结

在本文中，我们介绍了如何在 Pandas 数据框中选择列。我们学习了如何选择单个列，多个列和连续的列。这是 Pandas 数据框处理中的常见操作，对于数据分析和处理非常重要。