python如何一行一行录入

Python中可以使用多种方法来实现一行一行录入数据、使用input()函数、使用文件读取函数、使用迭代器和生成器等。 在本文中，我们将详细探讨这些方法，并提供实际的示例代码，以便更好地理解如何在Python中进行一行一行的录入操作。

一、使用input()函数

基本用法

在Python中，最简单的方式之一是使用input()函数来获取用户输入。这种方法非常适合在交互式程序中使用，尤其是在需要逐行输入数据的情况下。

while True:
    line = input("请输入一行数据（或输入'退出'以结束）：")
    if line.lower() == '退出':
        break
    print(f"你输入的是：{line}")

解释

上述代码每次循环都会提示用户输入一行数据，并将其存储在变量line中。如果用户输入“退出”，程序将结束循环。否则，它会继续提示用户输入下一行数据。

优点与局限性

优点：简单易用，适合初学者。
局限性：只能在交互式环境下使用，不适合处理大批量数据。

二、从文件中逐行读取

使用open()函数

基本用法

当需要从文件中逐行读取数据时，可以使用open()函数结合readline()或readlines()方法。

with open('example.txt', 'r', encoding='utf-8') as file:
    for line in file:
        print(line.strip())

解释

该代码使用with open语句打开一个名为example.txt的文件，并逐行读取文件内容。strip()方法用于删除每行末尾的换行符。

使用文件对象作为迭代器

文件对象本身是迭代器，因此可以直接在for循环中使用。

with open('example.txt', 'r', encoding='utf-8') as file:
    for line in file:
        print(line.strip())

优点与局限性

优点：适用于处理大文件，可以逐行读取，节省内存。
局限性：需要文件存在且路径正确。

三、使用迭代器和生成器

使用生成器逐行读取

基本用法

生成器是Python中一种强大的工具，特别适用于需要逐行处理数据的场景。

def read_lines(filename):
    with open(filename, 'r', encoding='utf-8') as file:
        for line in file:
            yield line.strip()
for line in read_lines('example.txt'):
    print(line)

解释

上述代码定义了一个生成器函数read_lines()，该函数逐行读取文件并使用yield关键字返回每一行。使用生成器的好处是它能够在处理大文件时节省内存，因为它不会一次性将所有内容读入内存。

使用itertools模块

基本用法

itertools模块提供了许多用于操作迭代器的工具。虽然它不是专门为逐行读取设计的，但可以结合其他方法使用。

import itertools
with open('example.txt', 'r', encoding='utf-8') as file:
    for line in itertools.islice(file, 0, None):
        print(line.strip())

解释

上述代码使用itertools.islice()函数从文件中按行读取数据。islice()函数的参数指定从第0行开始读取，直到文件结束。

优点与局限性

优点：生成器和迭代器能够高效地处理大数据集，节省内存。
局限性：代码相对复杂，初学者可能需要一些时间来掌握。

四、使用pandas读取大文件

基本用法

pandas是一个非常强大的数据处理库，适用于处理结构化数据。虽然它通常用于读取整个文件，但也可以逐行读取大文件。

import pandas as pd
chunk_size = 10  # 每次读取10行
for chunk in pd.read_csv('example.csv', chunksize=chunk_size):
    for line in chunk.itertuples():
        print(line)

解释

上述代码使用pandas.read_csv()函数按块读取CSV文件，每次读取10行数据。然后使用itertuples()方法逐行处理每个块的数据。

优点与局限性

优点：适用于处理大规模结构化数据，功能强大。
局限性：需要安装pandas库，代码相对复杂。

五、使用sys.stdin读取输入流

基本用法

在某些情况下，尤其是在处理脚本和命令行工具时，可以使用sys.stdin逐行读取输入流。

import sys
for line in sys.stdin:
    print(line.strip())

解释

上述代码从标准输入流中逐行读取数据，非常适合用于命令行工具或脚本中。

优点与局限性

优点：适用于命令行工具和脚本，灵活性高。
局限性：需要从标准输入流中获取数据，可能不适合所有场景。

六、使用第三方库

基本用法

除了pandas，还有许多第三方库可以用于逐行读取数据。例如，csv库非常适合读取CSV文件。

import csv
with open('example.csv', newline='', encoding='utf-8') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(', '.join(row))

解释

上述代码使用csv.reader()逐行读取CSV文件，并将每一行的数据以逗号分隔的形式打印出来。

优点与局限性

优点：适用于特定格式的文件，如CSV，功能强大。
局限性：需要安装和使用特定的第三方库。

七、使用内存映射文件

基本用法

对于非常大的文件，可以使用内存映射文件（memory-mapped file）进行高效的逐行读取。

import mmap
with open('example.txt', 'r+') as f:
    mmapped_file = mmap.mmap(f.fileno(), 0)
    for line in iter(mmapped_file.readline, b""):
        print(line.decode('utf-8').strip())
    mmapped_file.close()