python - 从python中的csv文件中获取特定列

Question

我是 python 编程新手，我有一个关于读取 csv 文件的问题。我想做的是将我的 csv 文件分成 2 个并保存在两个列表中。因此，形成我的下一个 csv 文件:

    5.1,3.5,1.5,0.2,setosa
    4.9,3.0,1.4,0.2,setosa
    4.6,3.1,1.5,0.2,setosa

我需要我的列表看起来像:

firstList = [ [5.1, 3.5, 1.5, 0.2], [4.9, 3.0, 1.4, 0.2], [4.6, 3.1, 1.5, 0.2] ]
secondList = ['setosa', 'setosa', 'setosa']

我尝试了不同的时间，但最接近的结果是下一个:

firstList = []
secondList = []

with open('file.csv') as file:
    fileReader = csv.reader(file, delimiter=',')
    for row in fileReader:
        firstList.append(row)
        secondList.append(row[4])

这个的输出是:

firstList = [ ['5.1', '3.5', '1.5', '0.2', 'setosa'], ['4.9', '3.0', '1.4', '0.2', 'setosa'], ['4.6', '3.1', '1.5', '0.2', 'setosa'] ]
secondList = ['setosa', 'setosa', 'setosa']

我只想从我的第一个列表中删除字符串“setosa”，并将其他所有内容转换为 int任何帮助将不胜感激。谢谢!

Answer 1

您可以构建该行中前三个单元格的子列表，然后在根据它们创建新列表时构建 float 。

firstList = []
secondList = []

with open('file.csv') as file:
    fileReader = csv.reader(file, delimiter=',')
    for row in fileReader:
        firstList.append([float(cell) for cell in row[:3]])
        secondList.append(row[4])