python 筛选符合条件的行下标

原创

mob64ca12d26eb9 2024-03-30 05:40:04 ©著作权

文章标签 Python 数据集代码示例 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d26eb9的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python中筛选符合条件的行下标

在数据处理和分析中，我们经常需要筛选出符合特定条件的数据行。在Python中，我们可以使用简洁而高效的方法来实现这一目标。本文将介绍如何使用Python筛选符合条件的行下标，以及相关的代码示例。

如何筛选符合条件的行下标

在Python中，我们可以使用列表推导式和内置函数来筛选符合条件的行下标。具体步骤如下：

遍历数据集的每一行；
判断该行是否符合设定的条件；
如果符合条件，则将该行的下标添加到结果列表中。

下面是一个简单的示例，假设我们有一个包含学生成绩的数据集，我们想要筛选出分数大于80分的学生的行下标：

scores = [75, 82, 90, 68, 88, 79, 95]
passed_indexes = [index for index, score in enumerate(scores) if score > 80]

print(passed_indexes)

运行以上代码，我们将得到结果 [1, 2, 4, 6]，这些数字表示符合条件的行的下标。

代码示例

下面我们来看一个更具体的示例，假设我们有一个包含多个学生信息的数据集，每一行包括学生的姓名、年龄和分数，我们希望筛选出年龄大于18岁且分数大于70分的学生的行下标：

students = [
    ["Alice", 20, 85],
    ["Bob", 17, 65],
    ["Charlie", 19, 75],
    ["David", 21, 70],
    ["Eve", 18, 80]
]

selected_indexes = [index for index, student in enumerate(students) if student[1] > 18 and student[2] > 70]

print(selected_indexes)

运行以上代码，我们将得到结果 [0, 4]，这表示第一位和第五位学生符合条件。