如何修复"row in csv_reader"不起作用？

Question

我正在读取一个长度可变的两列.csv文件。我编写了一些代码，应该能够将数据栏的第一行读取为 x 数据列和 y 数据列。这是代码：

def csvReader(filename):
with open(filename) as csvFile:
csvReader = csv.reader(csvFile, delimiter = ',')
rowCount = sum(1 for row in csvReader)
xData = np.zeros(rowCount)
yData = np.zeros(rowCount)
line_count = 0
firstLine = True
for row in csvReader:
print(row)
if firstLine:
firstLine = False
continue
xData[line_count] = row[0]
yData[line_count] = row[1]
line_count += 1
return xData,yData

它输出一个零数组，控制台从不显示任何打印输出，这似乎意味着整个 for 循环被跳过了。有关此问题的任何帮助将不胜感激。

Answer 1

当你这样做时，你会耗尽迭代器

rowCount = sum(1 for row in csvReader)

您需要倒带文件才能再次读取。

csvFile.seek(0)