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