訪問csv文件的第N行的最佳方法

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 == N: 
     print("This is the line.") 
     print(row) 
     break 

您還可以使用itertools.islice這是專爲這種類型的場景 - 訪問一個可迭代的特定片段，而不必讀取整個事情到內存中。它應該比循環不需要的行更高效。

with open(path, 'r') as f: 
    N = int(input('What line do you need? > ')) 
    print("This is the line.") 
    print(next(itertools.islice(csv.reader(f), N, None))) 

但是如果你的CSV文件很小，只是讀了整個事情到列表中，您可以再與以正常方式的索引來訪問。這也有一個好處，即您可以隨機訪問多個不同的行，而無需重置csv閱讀器。

my_csv_data = list(reader) 
print(my_csv_data[N]) 

Answer 2

你可以你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 3

您的解決方案其實並沒有那麼糟糕。推進文件迭代器到你想要的行是一個很好的方法，並在這樣的許多情況下使用。

如果你想讓它更簡潔，雖然，你可以使用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 4

你可以簡單地做：

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

Answer 5

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：

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

甚至更​​好通過不加載整個文件到內存中使用較少的內存現在你可以通過它的獲得任何行directly：

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

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

Answer 6

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) 

P.S.對於好奇，我最初的反應—也可以工作（可以說是更好）—是：

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