1000字范文,内容丰富有趣,学习的好帮手!
1000字范文 > python生成csv文件带水印_如何用python从生成器写入.csv文件

python生成csv文件带水印_如何用python从生成器写入.csv文件

时间:2018-08-26 10:11:09

相关推荐

python生成csv文件带水印_如何用python从生成器写入.csv文件

您的x已经是一个序列,不需要将其包装在列表中:for x in g:

writes.writerow(x)

您可以直接将生成器传递给writes.writerows():with open(f2, "w") as file1:

writes = csv.writer(file1, delimiter=' ', quoting=csv.QUOTE_ALL)

writes.writerows(mygen(reader))

演示:>>> import sys

>>> import csv

>>> def mygen(reader):

... for row in reader:

... yield row[0],row[1],row[2],row[3],row[4]

...

>>> reader = csv.reader('''\

... foo,bar,baz,spam,ham,eggs

... 42,81,13,97,73,100

... '''.splitlines())

>>> writes = csv.writer(sys.stdout, delimiter=' ', quoting=csv.QUOTE_ALL)

>>> writes.writerows(mygen(reader))

"foo" "bar" "baz" "spam" "ham"

"42" "81" "13" "97" "73"

你需要注意的是,你的reader并不是一个你已经筋疲力尽的生成器。如果它是一个csv.reader()对象,并且您以前读过所有行,则需要将底层文件对象重新卷绕回以fileobj.seek(0)开头的位置。不过,最好不要读两次文件。

如果reader对象生成序列(如列表或元组),则只需使用切片来获取前5个元素,无需键入每个索引:def mygen(reader):

for row in reader:

yield row[:5]

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。