Единственный способ, с помощью которого я использовал Python csv.reader, - это цикл for, который проходит через весь файл без сохранения прошлых значений read в переменных. Мне нужно работать только с двумя последовательными строками (огромного) файла за раз. Используя цикл csv.reader для цикла, я имею только 1 строку за раз.
Есть ли способ использовать модуль csv Python для ввода только одной строки файла csv без необходимости заканчивать чтение файла до конца?
Мне нужно установить переменные в значения в первой строке, установить второй набор переменных в значения следующей строки, использовать два набора переменных одновременно для вычислений, а затем переписать первый набор переменных на второй установить и прочитать новую строку, чтобы перезаписать второй набор.
Нет ничего, что заставило бы вас использовать читателя в цикле. Просто прочитайте первую строку, затем прочитайте вторую строку.
import csv
rdr = csv.reader(open("data.csv"))
line1 = rdr.next() # in Python 2, or next(rdr) in Python 3
line2 = rdr.next()
Если вы всегда смотрите ровно на две последовательные строки, это звучит для меня так, как будто вы могли бы воспользоваться парой recipe, Из модуля itertools:
from itertools import tee, izip
def pairwise(iterable):
"s -> (s0,s1), (s1,s2), (s2, s3), ..."
a, b = tee(iterable)
next(b, None)
return izip(a, b)
Вы использовали бы это так:
for first_dict, second_dict in pairwise(csv.DictReader(stream)):
# do stuff with first_dict and second_dict
Считать CSV:
readCSV = csv.reader(csvFile, delimiter=',')
Прочитайте следующую строку в Python 2.7:
row = readCSV.next()
Прочитайте следующую строку в Python 3.4:
row = readCSV.__next__()
Похищение воровства от TK...... в основном остается вопрос, что OP хочет делать с первой и последней строками файла?
prevLine = None
for x in csv.DictReader(stream):
if prevLine is not None:
DoWork(prevLine, x)
else:
Initialize(x)
prevLine = x
Finalize(prevLine)
Очевидным ответом является просто сохранение предыдущей строки на каждой итерации.
>>> for x in csv.DictReader(stream):
... print prevLine
... print x
... prevLine = x
....