pandas dataframe选择列|极客教程

pandas dataframe选择列

在Pandas中，DataFrame是一种用于存储和处理数据的二维标记数组。有时候我们需要从DataFrame中选择特定的列进行分析或操作。本文将详细介绍如何在Pandas中选择列，并对常用的方法进行演示。

读取数据创建DataFrame

首先，我们需要导入Pandas库并创建一个DataFrame来演示如何选择列。在本例中，我们将使用一个包含学生姓名、年龄和成绩的数据集作为示例。

import pandas as pd

# 创建一个包含学生信息的字典
data = {
    'Name': ['Alice', 'Bob', 'Charlie', 'David', 'Emily'],
    'Age': [20, 21, 22, 23, 24],
    'Score': [85, 90, 88, 87, 89]
}

# 创建DataFrame
df = pd.DataFrame(data)

print(df)

运行以上代码将得到以下输出：

      Name  Age  Score
0    Alice   20     85
1      Bob   21     90
2  Charlie   22     88
3    David   23     87
4    Emily   24     89

我们已经成功创建了一个包含学生信息的DataFrame，接下来我们将演示如何选择列。

选择单列

要选择单列，我们可以使用方括号加上列名的方式来进行选择。下面是一些示例代码：

# 选择Name列
name_column = df['Name']
print(name_column)

运行以上代码将得到Name列的数据：

0      Alice
1        Bob
2    Charlie
3      David
4      Emily
Name: Name, dtype: object

除了使用方括号选择列外，我们还可以使用属性/列名的方式来选择列。示例代码如下：

# 使用属性选择Age列
age_column = df.Age
print(age_column)

运行以上代码将得到Age列的数据：

0    20
1    21
2    22
3    23
4    24
Name: Age, dtype: int64

选择多列

如果我们想选择DataFrame中的多列，可以将列名放在一个列表中进行选择。示例代码如下：

# 选择Name和Score列
name_score_columns = df[['Name', 'Score']]
print(name_score_columns)

运行以上代码将得到Name和Score列的数据：

      Name  Score
0    Alice     85
1      Bob     90
2  Charlie     88
3    David     87
4    Emily     89

选择连续的多列

有时候我们需要选择连续的多列，可以使用切片来选择列。示例代码如下：

# 选择Age和Score列
age_score_columns = df.loc[:, 'Age':'Score']
print(age_score_columns)

运行以上代码将得到Age和Score列的数据：

   Age  Score
0   20     85
1   21     90
2   22     88
3   23     87
4   24     89

选择列的不同方式

除了上面介绍的方法之外，Pandas还提供了一些其他选择列的方式。

使用.loc方法选择列

.loc方法可以根据标签选择特定的列。示例代码如下：

# 使用.loc选择Name列
name_column_loc = df.loc[:, 'Name']
print(name_column_loc)

使用.iloc方法选择列

.iloc方法可以根据位置选择特定的列。示例代码如下：

# 使用.iloc选择第一列
first_column = df.iloc[:, 0]
print(first_column)

使用.filter方法根据列名选择列

.filter方法可以根据列名的部分或全部匹配来选择列。示例代码如下：

# 使用.filter选择包含'me'的列
filtered_columns = df.filter(like='me')
print(filtered_columns)

总结

在Pandas中，选择列是非常常见的操作。通过本文的介绍，我们学会了如何选择单列、多列以及连续的多列，并了解了一些其他选择列的方法。选择列是数据分析和处理中的重要一环，熟练掌握选择列的方法可以提高我们的工作效率。

pandas dataframe选择列