笔记大全

Python读取CSV前几行

更新时间：2023-09-30

Python如何读取CSV文件前几行?

CSV（Comma Separated Values）格式是一种常见的数据格式，常用于将表格数据导出到其他软件中。在Python中，我们可以使用csv模块来读取和处理CSV文件。通常情况下，我们需要读取文件中的前几行来进行预览或者筛选。下面我们将介绍如何使用Python读取CSV文件前几行。

1.使用csv.reader()和next()

使用csv.reader()函数来读取CSV文件，并使用next()函数来跳过第一行表头。然后循环读取指定行数的数据。

import csv

# 打开CSV文件
with open('example.csv') as file:
    # 创建CSV读取器对象
    reader = csv.reader(file)
    # 跳过第一行表头
    next(reader)
    # 读取前5行数据
    for i in range(5):
        try:
            row = next(reader)
            print(row)
        except StopIteration:
            break

在上面的代码中，我们使用了with语句来打开CSV文件，并将文件指针赋给file变量。然后，我们创建了一个csv.reader对象，并使用next()函数跳过了第一行表头。

接着，我们使用for循环来循环读取指定行数的数据。这里使用了try/except语句来捕获StopIteration异常，因为当读取到文件结尾的时候，next()函数将会抛出该异常。

2.使用pandas库的read_csv()方法

pandas是Python中一个强大的数据分析库，它提供了很多方便的函数来读取和处理各种数据文件。在读取CSV文件的时候，我们可以使用pandas库中的read_csv()方法。read_csv()方法默认读取整个文件，但是我们可以通过设置nrows参数来控制读取的行数。

import pandas as pd

# 读取前5行数据
df = pd.read_csv('example.csv', nrows=5)
print(df)

在上面的代码中，我们使用了pandas库的read_csv()方法来读取CSV文件，并通过nrows参数来设置读取的行数。读取到的数据保存在一个DataFrame对象中，并打印出来。

3.使用numpy库的genfromtxt()方法

numpy是Python中一个广泛使用的科学计算库，其中包含了很多方便的函数和方法来处理数据。我们可以使用numpy库中的genfromtxt()方法来读取CSV文件的前几行数据。

import numpy as np

# 读取前5行数据
data = np.genfromtxt('example.csv', delimiter=',', skip_header=1, max_rows=5)
print(data)

在上面的代码中，我们使用了numpy库的genfromtxt()方法来读取CSV文件，并设置了delimiter参数为逗号，skip_header参数为1，max_rows参数为5。这样我们就可以读取到CSV文件的前五行数据，并打印出来。

总结

在Python中，我们可以使用csv模块、pandas库和numpy库来读取CSV文件的前几行数据。使用csv模块需要手动跳过表头，并循环读取指定行数的数据。使用pandas库的read_csv()方法可以方便地读取指定行数的数据，并返回一个DataFrame对象。使用numpy库的genfromtxt()方法也可以读取指定行数的数据，并返回一个二维数组。

c语言编程笔录