python-仅解压缩CSV阅读器的前几列？

读取CSV文件时：

with open("foo.csv") as foo:
    for (a,b) in csv.reader(foo):
        ....

我得到错误

ValueError: too many values to unpack

当文件包含多于两列时.

有没有办法拆开前两列并忽略其余列？

我想我能做

with open("foo.csv") as foo:
    for row in csv.reader(foo):
        a,b = row[0:2]
        ....

但这看起来很丑.

PS.我正在使用python2,如果那很重要的话.

解决方法:

使用生成器：

with open("foo.csv") as foo:
    for a,b in (r[0:2] for r in csv.reader(foo)):
         ...

这可以更清楚地显示您的意图.它基本上等效于您的“丑陋”方式,但是它看起来容易得多.

标签：iterable-unpacking,csv,python
来源： https://codeday.me/bug/20191121/2052420.html