python - 如何在 python 中循环遍历 .csv 文件

Question

我正在尝试遍历 .csv 文件以便读取每一行，但我收到以下错误:

AttributeError: '_io.TextIOWrapper' 对象没有属性 'iterrows'

到目前为止，这是我的代码:

def i_list(input_file):

    input_list = []
    file = open(input_file, 'r')
    for i, j in file.iterrows():
            num1 = j[1]
            num2 = j[2]
            input_list.append(num1)
            input_list.append(num2)

i_list('nums.csv')

Answer 1

您可以使用 csv library

import reader from csv

input_list=[]
f = reader(open(input_file))
header = next(f) # if there is a header
for num1, num2 in f:
    input_list.append(num1)
    input_list.append(num2)