python - 如何在循环中写入csv文件

Question

我有一段执行循环的代码：

for i in os.listdir(os.getcwd()):
   if i.endswith(".gz"):
        myfilename=i
        getHum=int(subprocess.check_output('''zcat %s |  awk 'BEGIN {FS=";"};{print $6}' | grep -i 'hummer' | wc -l''' % myfilename, shell=True))
        getBMW=int(subprocess.check_output('''zcat %s |  awk 'BEGIN {FS=";"};{print $6}' | grep -i 'bmw' | wc -l''' % myfilename, shell=True))
        getAudi=int(subprocess.check_output('''zcat %s |  awk 'BEGIN {FS=";"};{print $6}' | grep -i 'audi' | wc -l''' % myfilename, shell=True))

现在，我正在尝试修改它，因此我将输出写入逗号 sep csv 文件。 如何遍历每个文件并将提取的内容写入 csv 文件？ 谢谢。

Answer 1

我认为这样的事情会做你想要的：

def get_brand_count(filename, brand):
    return int(subprocess.check_output('''zcat %s |  '''
                                       '''awk 'BEGIN {FS=";"};{print $6}' | '''
                                       '''grep -i %r | wc -l''' %
                                       (filename, brand), shell=True))

brands = 'hummer', 'bmw', 'audi'
with open('results.csv', 'wb') as csvfile:
    csvwriter = csv.writer(csvfile)
    csvwriter.writerow(brands)  # csv header row (if desired)
    for myfilename in glob.iglob('*.gz'):
        csvwriter.writerow([get_brand_count(myfilename, brand)
                                for brand in brands])