如何修复“ csv_reader中的行”不起作用？

Question

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

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)