python - 如何将 for 循环输出写入 csv 文件？

Question

我有一个返回列表的函数 ..

def give_list(number, type, index): 
        list_ = []
        do something 
    return list_

此函数的输出格式如下；

['Training/1.png', 3450, 3865, 1517, 1789, 'GOOD']

现在我使用 for 循环获取更多列表，如上所示，并写入 .csv 文件..

for key, value in dic_.items():
    list_ = give_list(key, "CC", 2)
    df = pd.DataFrame(list_)
    df.to_csv('/path/to/write/testFile.csv', sep=',') 

例如我的循环生成以下输出；

['Training/1.png', 1255, 4556, 1455, 1789, 'GOOD']
['Training/2.png', 4555, 2899, 6989, 1586, 'BAD']
['Training/3.png', 4744, 2478, 9862, 1755, 'BAD']
['Training/4.png', 3450, 6989, 2455, 3866, 'GOOD']

for 循环内的 df.to_csv('/path/to/write/testFile.csv', sep=',') 无法帮助我将输出写入 .csv 文件。我怎样才能做到这一点？我需要在 .csv 文件中看到这样的输出；

'Training/1.png', 1255, 4556, 1455, 1789, 'GOOD'
'Training/2.png', 4555, 2899, 6989, 1586, 'BAD'
'Training/3.png', 4744, 2478, 9862, 1755, 'BAD'
'Training/4.png', 3450, 6989, 2455, 3866, 'GOOD'

Answer 1

试试这个:

import csv 
payload = [give_list(key, "CC", 2) for key in dic_] # create list of lists(rows) payload
with open('/path/to/write/testFile.csv', "w", newline="") as f:
    writer = csv.writer(f)
    writer.writerows(payload)