使用python

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])