如何為CSV編寫器編寫python函數

Question

我有下面的代碼工作正常：

with open('userWithAgentProp.csv','w+') as f:
        w = csv.DictWriter(f,user_keys)
        w.writeheader()
        for user in userAgentProp_list:
            w.writerow(user)

我使用以下代碼編寫了一個函數：

def createCSVOutput(fileName, keys, listOfLists):
    with open(fileName, 'w+') as f:
        w= csv.dictWriter(f, keys)
        w.writeheader()
        for row in listOfLists:
            w.writerow(row)

當我調用該函數時：

createCSVOutput('new_test_csv.csv', user_keys, userAgentProp_list)

我收到以下錯誤：

File "mongodb_script_2.py", line 101, in <module>
  createCSVOutput('new_test_csv.csv', user_keys, userAgentProp_list)
File "mongodb_script_2.py", line 54, in createCSVOutput
  w= csv.dictWriter(f, keys)
AttributeError: 'module' object has no attribute 'dictWriter'

為什么它對腳本中的user_keys變量有效，但對函數不起作用？

Answer 1

您有錯字：

csv.dictWriter是錯誤的。 它應該是csv.DictWriter 。