我正在尝试批量读取csv文件,并通过某些回调函数进行处理。
假设CSV文件有10行(没有标题),批处理大小为3。预期结果应该是4个批次。前3个批次有3行,第四个批次只包含1行。 我编写的代码只能生成3个批次。如果批处理大小为1/2/5/10,则一切正常。
import csv
with open('file.csv', 'r') as csvfile:
reader = csv.reader(csvfile)
header = next(reader) # skip header
batch_size = 3
batch = []
count = 0
for row in reader:
if count >= batch_size:
do_something(batch)
batch = []
count = 0
batch.append(row)
count += 1
假设CSV文件有10行(没有标题),批处理大小为3。预期结果应该是4个批次。前3个批次有3行,第四个批次只包含1行。 我编写的代码只能生成3个批次。如果批处理大小为1/2/5/10,则一切正常。