gpt4 book ai didi

python - 使用一个命令将 csv 读入列表,然后读入

转载 作者:行者123 更新时间:2023-11-30 23:36:16 25 4
gpt4 key购买 nike

是否可以像这样读取 csv:

1,22,5
1,33,9
3,44,5
4,32,5
5,53,5
6,64,5
7,12,7

使用一个命令访问格式为[(1,22,5),...]的列表?

之后我如何在列表中搜索以获取例如所有第一列为 1 的元素?

例如:get all with row1 == 1 将返回

1,22,5
1,33,9

最佳答案

import csv

with open('data.csv', 'rb') as f:
rows = [[int(el) for el in row] for row in csv.reader(f)]


>>> rows
[[1, 22, 5], [1, 33, 9], [3, 44, 5], [4, 32, 5], [5, 53, 5], [6, 64, 5], [7, 12, 7]]
>>> [row for row in rows if row[0] == 1]
[[1, 22, 5], [1, 33, 9]]

关于python - 使用一个命令将 csv 读入列表,然后读入,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/16648269/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com