Python中从多个文本文件中提取指定行的简便方法

在Python中，我们可以使用多种方法从多个文本文件中提取指定行。下面是几种简便的方法及其使用示例：

方法一：使用fileinput模块

fileinput模块使得我们能够轻松地迭代多个文件中的所有行，然后我们可以在循环中检查每一行是否符合我们的要求。

import fileinput

# 提取包含指定关键词的行
def extract_lines(keyword, files):
    for line in fileinput.input(files):
        if keyword in line:
            print(line)

# 提取指定行号的行
def extract_lines(line_numbers, files):
    for i, line in enumerate(fileinput.input(files)):
        if i+1 in line_numbers:
            print(line)

# 使用示例1：提取包含指定关键词的行
extract_lines("Python", ["file1.txt", "file2.txt"])

# 使用示例2：提取指定行号的行
extract_lines([3, 5, 7], ["file1.txt", "file2.txt"])

方法二：使用正则表达式

如果我们需要根据更具体的模式来提取行，可以使用正则表达式。

import re

# 提取满足指定正则表达式模式的行
def extract_lines(pattern, files):
    for file in files:
        with open(file, "r") as f:
            for line in f:
                if re.search(pattern, line):
                    print(line)

# 使用示例：提取满足指定正则表达式模式的行
extract_lines(r"\b\d{4}-\d{2}-\d{2}\b", ["file1.txt", "file2.txt"])

方法三：使用pandas库

如果我们的文本文件以表格形式存储，并且需要提取特定行，可以使用pandas库。

import pandas as pd

# 提取指定行号的行
def extract_lines(line_numbers, files):
    for file in files:
        df = pd.read_csv(file)
        lines = df.iloc[line_numbers]
        print(lines)

# 使用示例：提取指定行号的行
extract_lines([3, 5, 7], ["file1.csv", "file2.csv"])

方法四：使用numpy库

如果我们的文本文件已经以数组形式存储，并且需要提取特定行，可以使用numpy库。

import numpy as np

# 提取指定行号的行
def extract_lines(line_numbers, files):
    for file in files:
        arr = np.loadtxt(file)
        lines = arr[line_numbers]
        print(lines)

# 使用示例：提取指定行号的行
extract_lines([3, 5, 7], ["file1.txt", "file2.txt"])

请根据您的具体需求选择适合的方法来提取指定行的文本文件内容。这些方法都可以根据您的需要进行修改和扩展。