python - 向 .csv 写入多于一列

Question

我有这个代码:

#//Running my argument to check if the item in list is in dictionary and returning    this + corresponding value in dictionary//#
for key in n:
    if DICT.get(key):
        print ((key) + ' : ' + DICT[key])
    else:
        print((key) + ' : ' + "Not Available")

#Writing to .cvs file       
with open('test_write.csv', 'w') as fp:
    a = csv.writer(fp)
    a.writerows(n)

我想将上面 for 循环的结果写入.csv。
目前，我将代码第一部分的每个字母放在单独的列中。
我需要在每一列中都有两个..
我建议我需要将 for 循环转换为字典？但我可能是错的......

关于如何最简单地做到这一点有什么想法吗？

编辑:
使用您的建议:

    with open('test_write.csv', 'w') as fp:
        a = csv.writer(fp)
        for key in n:
        if DICT.get(key):
            print ((key) + ' : ' + DICT[key])
            a.writerow(n)
        else:
            print((key) + ' : ' + "Not Available")
            a.writerow(DICT.get(n,"Not Available") for name in n)

我没有得到我所期望的
我得到了我的列表 n 的 4 行。DICT 中没有显示该值。我做错了什么...

Answer 1

writerows 接受一个可迭代列表。尝试使用 writerow。从表面上看，n 是一个字典，因此要获取一行标题和一行值，请执行以下操作:

a.writerow(n.keys())
a.writerow(n.values())

您可以为第一行编写a.writerow(n)，但我更喜欢更明确地展示这一点。

还有一个添加所有默认值的小快捷方式:

names = ['list','of','all','expected','keys']
data = {'list':'A', 'of':'zoo', 'keys':'foo'}
default = dict.fromkeys(names,"Not Available")
default.update(data)
data = default

留下数据和内容:

{'all': 'Not Available', 'of': 'zoo', 'list': 'A', 'expected': 'Not Available', 'keys': 'foo'}

编辑:

给定 DICT = {1:a, 2:b, 3:c, 4:d} 和列表 n = [1, 2, 5, 6]，只需执行以下操作:

a.writerow(n)
a.writerow(DICT.get(name,"Not Available") for name in n)

将在 CSV 文件中打印两行，其中一行包含 n 中的键名称，另一行包含 DICT 中的值，如果特定键不在 DICT 中，则打印“不可用”字样。

编辑2:

你太努力了 - DICT.get 将处理条目是否存在:

with open('test_write.csv', 'w') as fp:
    a = csv.writer(fp)
    a.writerow(n)
    a.writerow(DICT.get(name,"Not Available") for name in n)

这是相同的代码，但形式更详细:

with open('test_write.csv', 'w') as fp:
    a = csv.writer(fp)
    # write row of header names
    a.writerow(n)

    # build up a list of the values in DICT corresponding to the keys in n
    values = []
    for name in n:
        if name in DICT:
            values.append(DICT[name])
        else:
            values.append("Not Available")
    # or written as a list comprehension:
    # values = [DICT[name] if name in DICT else "Not Available" for name in n]
    # 
    # or just use DICT.get, which does the name checking for us, and chooses the 
    # default value if the key is not found
    # values = [DICT.get(name, "Not Available") for name in n]

    # now write them out
    a.writerow(values)

    # or finally, the list build and writerow call all in one line
    # a.writerow(DICT.get(name,"Not Available") for name in n)

编辑:

    # to write out the transpose of the previous lines (that is, instead of 
    # a line of names and a line of the matching values, a line for each
    # name-value pair), use Python's zip builtin:
    for nameValueTuple in zip(n,values):
        a.writerow(nameValueTuple)