Pandas数据筛选的高级技巧及实例应用-Python教程-PHP中文网

Pandas数据筛选的高级技巧及实例应用

PHPz

发布： 2024-01-24 08:28:06

原创

1540人浏览过

pandas数据筛选的高级技巧与应用案例

Pandas是一个强大的数据处理和分析工具，它提供了很多灵活的功能和方法，能够方便地对数据进行筛选和处理。本文将介绍几种Pandas数据筛选的高级技巧，并通过实际案例提供具体的代码示例。

一、基本数据筛选

Pandas提供了多种方法来对数据进行基本的筛选，如使用布尔索引、loc或iloc方法等。以下是一些常见的基本数据筛选案例。

布尔索引筛选

布尔索引可以用来根据某个条件筛选数据。例如，我们有一个包含学生信息的数据框，想筛选出成绩大于60分的学生。可以使用如下代码实现：

import pandas as pd

data = {
    '姓名': ['张三', '李四', '王五', '赵六'],
    '成绩': [80, 70, 90, 50]
}

df = pd.DataFrame(data)
df_filtered = df[df['成绩'] > 60]

print(df_filtered)

登录后复制

loc方法筛选

loc方法可以根据行标签和列标签进行数据筛选。例如，我们有一个包含学生信息的数据框，想筛选出姓名为张三和李四的学生的成绩和年龄。可以使用如下代码实现：

import pandas as pd

data = {
    '姓名': ['张三', '李四', '王五', '赵六'],
    '成绩': [80, 70, 90, 50],
    '年龄': [18, 19, 20, 21]
}

df = pd.DataFrame(data)
df_filtered = df.loc[df['姓名'].isin(['张三', '李四']), ['成绩', '年龄']]

print(df_filtered)

登录后复制

二、高级数据筛选

除了基本数据筛选方法外，Pandas还提供了许多高级的数据筛选技巧，如使用query方法、使用索引对象MultiIndex进行多层次筛选等。以下是几个案例说明。

query方法筛选

query方法可以通过类似SQL的语法来筛选数据。例如，我们有一个包含学生信息的数据框，想筛选出成绩大于60分并且年龄在18到20岁之间的学生。可以使用如下代码实现：

import pandas as pd

data = {
    '姓名': ['张三', '李四', '王五', '赵六'],
    '成绩': [80, 70, 90, 50],
    '年龄': [18, 19, 20, 21]
}

df = pd.DataFrame(data)
df_filtered = df.query('成绩 > 60 and 18 <= 年龄 <= 20')

print(df_filtered)

登录后复制

使用MultiIndex筛选

如果数据框有多层次的索引，可以利用MultiIndex对象进行多层次筛选。例如，我们有一个包含学生信息的数据框，其中索引包括班级和学号两层次，想筛选出1班学号为001和002的学生。可以使用如下代码实现：

import pandas as pd

data = {
    '姓名': ['张三', '李四', '王五', '赵六'],
    '成绩': [80, 70, 90, 50],
}

index = pd.MultiIndex.from_tuples([('1班', '001'), ('1班', '002'), ('2班', '001'), ('2班', '002')])
df = pd.DataFrame(data, index=index)
df_filtered = df.loc[('1班', ['001', '002']), :]

print(df_filtered)

登录后复制

三、案例分析

现在我们以一个真实的数据集为例，进一步说明Pandas数据筛选的高级技巧。假设我们有一个汽车销售数据集，其中包含了车辆品牌、车型、销售量和销售额等信息。我们想筛选出销售量超过1000辆且销售额超过100万的车型。以下是代码示例：

import pandas as pd

data = {
    '品牌': ['宝马', '奥迪', '奔驰', '大众'],
    '型号': ['X3', 'A6', 'E级', '朗逸'],
    '销售量': [1200, 800, 1500, 900],
    '销售额': [1200, 900, 1800, 800]
}

df = pd.DataFrame(data)
df_filtered = df.query('销售量 > 1000 and 销售额 > 1000000')

print(df_filtered)

登录后复制

通过以上代码，我们成功筛选出了销售量超过1000辆且销售额超过100万元的车型。

综上所述，Pandas提供了丰富的数据筛选功能和方法，从基本的布尔索引、loc和iloc方法到高级的query方法和MultiIndex筛选，可以满足不同场景下的数据筛选需求。以上案例展示了一些常见的数据筛选技巧和应用，希望可以对读者在实际应用中有所帮助。

以上就是Pandas数据筛选的高级技巧及实例应用的详细内容，更多请关注php中文网其它相关文章！