Python，使用从 function 返回的字典列表

Question

我正在尝试返回字典列表，然后在 function 之外使用它：

 def myfunc():
     fileName: some file on my system
     with open(fileName) as csv1:
         dataDict = csv.DictReader(csv1, delimiter=',')

         return dataDict    

但是当我调用 t function 时，我收到错误“ValueError: I/O operation on closed file”。

myDict = myfunc()

for row in myDict:
    print(row)

我应该如何声明和使用字典列表？ 返回字典后，我还需要访问其字段名。 非常感谢任何提示或帮助。

Answer 1

您可以将其复制到另一个变量中并返回...

 def myfunc():
     fileName: some file on my system
     mydict = {}
     with open(fileName) as f: 
         for row in csv.DictReader(f, delimiter=','):
             mydict[row['first_column']] = row['second_column'] + row['third_column']

     return mydict  

将列名替换为 csv 中的列名。

Answer 2

DictReader 很懒惰。 实际上把它变成一个列表并返回它，以便文件在它仍然打开的时候被读取。

def myfunc():
     fileName: some file on my system
     with open(fileName) as csv:
         return list(csv.DictReader(csv, delimiter=','))

Answer 3

with语句自动关闭文件。 当您尝试通过with语句之外的 DictReader 访问数据时，DictReader 无法访问已关闭的文件，因此您会收到 IOError。

在从 function 返回之前，您可以通过将文件中的行收集到列表中来解决此问题：

 def myfunc():
     fileName: some file on my system
     with open(fileName) as csv1:
         dataDict = csv.DictReader(csv1, delimiter=',')
         list_of_dicts = list(datadict)
         return list_of_dicts