我是 Python 的新手,我只想打印一个巨大的 csv 文件的前 10 行。
到目前为止,这是我的代码,用于打印 csv 文件中的所有行
import csv
with open('titanic.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
print(row['survived'], row['pclass'], row['name'], row['sex'], row['age'])
最佳答案
使用itertools.islice
:
import csv
from itertools import islice
with open('titanic.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in islice(reader, 10): # first 10 only
print(row['survived'], row['pclass'], row['name'], row['sex'], row['age'])
同时,您还可以使用 operator.itemgetter
使列变得更容易一些:
import csv
from itertools import islice
from operator import itemgetter
get_columns = itemgetter('survived', 'pclass', 'name', 'sex', 'age')
with open('titanic.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in islice(reader, 10): # first 10 only
print(*get_columns(row))
关于python - 如何使用 Python 仅打印 csv 文件的前 10 行?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37088512/