python - 使用python计算csv特定列中的行条目数

Question

假设我有一个文件 items.csv，其中包含 3 列数据，但每列中填充的行数不同，例如:

ITEM1,ITEM2,ITEM3,
DATA,DATA,DATA,
DATA,,DATA
,,DATA

即往下看，ITEM1列只有2行，ITEM2有1个条目，ITEM3有3个数据条目(“DATA”)。

如何为 python 2.7 脚本编写一个函数来打开 items.csv 并返回特定列中的行数？

提前致谢。

Answer 1

csv.DictReader 就可以了:

   import csv

   def get_number_of_rows(filename, header):
      with open(filename) as f:
         reader = csv.DictReader(f)
         return sum([bool(row[header]) for row in reader])