python数据处理之如何选取csv文件中某几行的数据

下面是详细讲解“python数据处理之如何选取csv文件中某几行的数据”的完整攻略。

第一步：读取csv文件

要选取csv文件中的某几行数据，首先需要将这个csv文件读入到Python程序中。使用Python自带的csv模块可以轻松实现csv文件的读取和数据的处理。

import csv

with open('example.csv') as csv_file:
    csv_reader = csv.reader(csv_file)

在这段代码中，我们使用了built-in函数 open 以及 csv 模块中的 csv.reader 函数打开并处理csv文件。其中 csv.reader 函数返回的 csv_reader 是一个迭代器，可以用于逐行读取csv文件中的数据。

第二步：选取指定行的数据

选取指定行的数据，需要通过迭代csv_reader行数来进行判断并执行操作。

选取固定的行

如果想要选取csv文件中的某几行，可以使用 if 或者 for 循环来实现。

import csv

with open('example.csv') as csv_file:
    csv_reader = csv.reader(csv_file)
    line_count = 0
    for row in csv_reader:
        if line_count in [1, 3, 5]:
            print(f'{row[0]} works in the {row[1]} department, and earns {row[2]} dollars.')
        line_count += 1

在这段代码中，我们用到了列表 [1, 3, 5]，该列表包含csv文件中需要选取的行数。在循环中，如果 line_count 等于该列表中的任意一个元素，就会输出该行所代表的数据。

选取符合条件的行

如果想要选取符合某些条件的行，可以使用 if 条件判断来实现。

import csv

with open('example.csv') as csv_file:
    csv_reader = csv.reader(csv_file)
    for row in csv_reader:
        if row[1] == 'Engineering':
            print(f'{row[0]} works in the {row[1]} department, and earns {row[2]} dollars.')

在这段代码中，我们只选取了csv文件中部门为 "Engineering" 的行，输出了满足条件的人的名字、部门和薪水。

第三步：完整代码实例

下面是一个完整的例子，该例子将读取csv文件并选取固定的行和符合条件的行输出。

import csv

with open('example.csv') as csv_file:
    csv_reader = csv.reader(csv_file)
    line_count = 0
    for row in csv_reader:
        if line_count in [1, 3, 5]:
            print(f'{row[0]} works in the {row[1]} department, and earns {row[2]} dollars.')
        if row[1] == 'Engineering':
            print(f'{row[0]} works in the {row[1]} department, and earns {row[2]} dollars.')
        line_count += 1

以上就是python数据处理之如何选取csv文件中某几行的数据的完整攻略，这个过程中我演示了两个样例。