Python 读文件指定行|极客笔记

Python 读文件指定行

在日常工作和学习中，我们经常需要读取文件中的数据。而有时候我们只需要读取文件中的某一行或某几行，而不是将整个文件全部读取。在Python中，我们可以通过一些方法来实现读取文件指定行的功能。本文将详细介绍如何在Python中读取文件指定行的方法和实现。

1. 读取文件指定行的常用方法

1.1 使用linecache模块

Python中的linecache模块提供了一个简单的接口来读取文件的指定行。它可以很方便地读取文件中指定行的内容，而不需要将整个文件读取到内存中。下面是使用linecache模块读取文件指定行的示例代码：

import linecache

def read_specific_line(filename, line_num):
    line = linecache.getline(filename, line_num)
    return line

filename = 'example.txt'
line_num = 5
line = read_specific_line(filename, line_num)
print(line)

在上面的示例中，我们定义了一个read_specific_line函数，它接受两个参数：文件名和要读取的行号。然后调用linecache.getline方法来读取指定行的内容。最后打印出读取到的内容。

运行上面的代码，如果example.txt文件中有至少5行内容，那么将会打印出第5行的内容。

1.2 使用fileinput模块

fileinput模块也可以用来读取文件中的指定行。它可以按行读取文件内容，并且支持从命令行或文件中读取数据。下面是使用fileinput模块读取文件指定行的示例代码：

import fileinput

def read_specific_line(filename, line_num):
    lines = fileinput.input(files=(filename,))
    for index, line in enumerate(lines, start=1):
        if index == line_num:
            return line

filename = 'example.txt'
line_num = 5
line = read_specific_line(filename, line_num)
print(line)

在上面的示例中，我们定义了一个read_specific_line函数，它和上面的方法类似，接受两个参数：文件名和要读取的行号。然后使用fileinput.input方法按行读取文件内容，并且使用enumerate方法来遍历每一行，并返回指定行的内容。

运行上面的代码，同样可以读取example.txt文件中的第五行内容并打印出来。

2. 示例

为了更好地理解如何读取文件中的指定行，我们可以创建一个example.txt文件，里面包含一些测试数据，如下所示：

Line 1
Line 2
Line 3
Line 4
Line 5
Line 6

接着，我们可以使用上面提到的两种方法来读取example.txt文件中的第五行，代码如下：

import linecache
import fileinput

def read_specific_line_linecache(filename, line_num):
    line = linecache.getline(filename, line_num)
    return line

def read_specific_line_fileinput(filename, line_num):
    lines = fileinput.input(files=(filename,))
    for index, line in enumerate(lines, start=1):
        if index == line_num:
            return line

filename = 'example.txt'
line_num = 5

line_linecache = read_specific_line_linecache(filename, line_num)
line_fileinput = read_specific_line_fileinput(filename, line_num)

print("Using linecache:")
print(line_linecache)

print("Using fileinput:")
print(line_fileinput)

运行上面的代码，将会输出如下结果：

Using linecache:
Line 5

Using fileinput:
Line 5

通过上面的示例，我们可以清楚地看到如何使用linecache和fileinput模块来读取文件中的指定行。这两种方法在读取文件中大量数据时非常方便和高效。当我们只需要读取文件中部分行的内容时，可以选择使用这两种方法。

总结一下，本文介绍了在Python中读取文件中指定行的两种常用方法：使用linecache模块和使用fileinput模块。通过这些方法，我们可以方便地读取文件中的指定行，并处理文件操作时更加灵活和便捷。

Python 读文件指定行