python - 如何获取 CSV 文件的行数？

Question

fieldnames = ['first_name', 'last_name', 'address']
with open('names.csv') as csvfile:
    reader = csv.DictReader(csvfile, fieldnames=fieldnames)
    for row in reader:
        print(row['first_name'], "'s number", row['last_name'], "address", row['adres'])

这是我打印 CSV 文件的代码。如果 CSV 文件是空的，我想打印它是空的。我想如果我能得到文件的行数，我就可以检查它是否为空。

Answer 1

使用sum 函数(使用生成器表达式)获取行数的有效方法:

with open('names.csv') as csvfile:
    row_count = sum(1 for row in csvfile)
    print(row_count if row_count else 'Empty')