迭代不能在Python中使用「打開」CSV模塊工作

Question

我想使用for創建2個或多個CSV文件，但只創建1個CSV。

import csv 
import os 

foods = ['pasta','chocolate'] 
for food in foods: 
    with open(food+'.csv', 'w') as csvfile: 
     fieldnames = ['id', 'Category', 'Name'] 

     writer = csv.DictWriter(csvfile, fieldnames=fieldnames) 
     writer.writeheader() 
     os.chdir('E:/data/'+food) 
     folders = os.listdir(os.curdir) 
     for filename in folders: 
       writer.writerow({'id':'', 'Category':food, 'Name':filename}) 

在這種情況下，只有pasta.csv創建成功。我想要創建chocolate.csv。 任何解決方案？謝謝

Answer 1

我懷疑是工作，問題出在os.chdir。你只需要它來獲取目錄列表，你可以考慮使用glob代替：

import csv 
from glob import glob 

foods = ['pasta','chocolate'] 
for food in foods: 
    with open(food+'.csv', 'w') as csvfile: 
     fieldnames = ['id', 'Category', 'Name'] 

     writer = csv.DictWriter(csvfile, fieldnames=fieldnames) 
     writer.writeheader() 
     folders = glob('E:/data/'+food+'/*/') 
     for filename in folders: 
       writer.writerow({'id':'', 'Category':food, 'Name':filename})