访问第 N 行 csv 文件的最佳方法

Question

我必须访问 CSV 文件中的第 N 行。

这就是我所做的：

import csv

the_file = open('path', 'r')
reader = csv.reader(the_file)

N = input('What line do you need? > ')
i = 0

for row in reader:
    if i == N:
        print("This is the line.")
        print(row)
        break

    i += 1

the_file.close()

...但这感觉不是最理想的。精确编辑：如果文件很大，我不想遍历所有行，也不想将整个文件加载到内存中。

我确实希望像 reader[N] 这样的东西存在，但我还没有找到。

编辑答案：这条线（来自选择的答案）是我正在寻找的：

next(itertools.islice(csv.reader(f), N, None)

Answer 1

您可以使用enumerate 遍历列表，直到找到正确的行：

for i, row in enumerate(reader):
    if i == line_number:
        print("This is the line.")
        print(row)
        break

您还可以使用专为此类场景设计的itertools.islice - 访问可迭代的特定切片而不将整个内容读入内存。它应该比遍历不需要的行更有效。

def get_csv_line(path, line_number):
    with open(path) as f:
        return next(itertools.islice(csv.reader(f), line_number, None))

但是，如果您的 CSV 文件很小，只需将整个内容读入一个列表，然后您就可以使用索引以正常方式访问该列表。这还有一个好处是您可以以随机顺序访问多个不同的行，而无需重置 csv 阅读器。

with open(path) as f:
    my_csv_data = list(csv.reader(f))
print(my_csv_data[line_number])

Answer 2

您的解决方案实际上并没有那么糟糕。将文件迭代器推进到您想要的行是一种很好的方法，并且在许多这样的情况下使用。

如果你想让它更简洁，你可以使用 next 和 enumerate 和 generator expression：

import csv

the_file = open('path', 'r')
reader = csv.reader(the_file)

N = int(input('What line do you need? > '))

line = next((x for i, x in enumerate(reader) if i == N), None)
print(line)

the_file.close()

其中的None 是如果找不到该行将返回的内容（N 太大）。不过，您可以选择任何其他值。

---

您也可以使用with-statement 打开文件以使其自动关闭：

import csv

with open('path', 'r') as the_file:
    reader = csv.reader(the_file)

    N = int(input('What line do you need? > '))

    line = next((x for i, x in enumerate(reader) if i == N), None)
    print(line)

---

如果你真的想缩小尺寸，你可以这样做：

from csv import reader
N = int(input('What line do you need? > '))
with open('path') as f:
    print(next((x for i, x in enumerate(reader(f)) if i == N), None))

Answer 3

itertools 模块有许多用于创建专用迭代器的函数——它的islice() 函数可以用来轻松解决这个问题：

import csv
import itertools

N = 5  # desired line number

with open('path.csv', newline='') as the_file:
    row = next(csv.reader(itertools.islice(the_file, N, N+1)))

print("This is the line.")
print(row)

附：出于好奇，我最初的反应——也有效（可以说更好）——是：

    row = next(itertools.islice(csv.reader(the_file), N, N+1))

Answer 4

你可以这样做：

n = 2 # line to print
fd = open('foo.csv', 'r')
lines = fd.readlines()
print lines[n-1] # prints 2nd line
fd.close()

或者通过不将整个文件加载到内存中来更好地利用更少的内存：

import linecache
n = 2
linecache.getline('foo.csv', n)

Answer 5

您可以将for 循环最小化为理解表达式，例如

row = [row for i,row in enumerate(reader) if i == N][0]  

# or even nicer as seen in iCodez code with next and generator expression

row = next(row for i,row in enumerate(reader) if i == N)

Answer 6

import csv
with open('cvs_file.csv', 'r') as inFile: 
    reader = csv.reader(inFile)
    my_content = list(reader)

line_no = input('What line do you need(line number begins from 0)? > ')
if line_no < len(my_content):
    print(my_content[line_no])
else:
    print('This line does not exists')

---

作为result，现在您可以通过index directly 获取任何线路：

What line do you need? > 2
['101', '0.19', '1']

What line do you need? > 100
This line does not exists