如何为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 。