1

我正在尝试使用以下代码将字典值列表提取到文件中：

import csv

def function(file, output, deli = ','):

    dictionary = dict()
    with open(file, 'r') as source, open(output, 'w') as outp:
        data = csv.reader(source)
        line0 = next(data)
        i = 0
        for element in line0:
            dictionary[i] = element
            i += 1
        my_writer = csv.writer(outp)
        for element in dictionary.values():
            print(element)
            my_writer.writerow(element)

if __name__ == '__main__':
    from sys import argv
    if len(argv) == 2:
        function(argv[1])
    elif len(argv) == 3:
        function(argv[1], argv[2])
    elif len(argv) == 4:
        function(argv[1], argv[2], argv[3])

    print("ok")

要在 shell 上运行此代码，我使用命令：python function.py input output

但是，尝试使用 csv 文件，例如：alpha、beta、gamma、delta

我得到以下结果：

a,l,p,h,a

 ,b,e,t,a

 ,g,a,m,m,a

 ,d,e,l,t,a

我尝试将分隔符更改为 ' '，但使用空格而不是逗号得到了相同的结果。

我错过了什么吗？

4

2 回答 2

2

问题出在线路上

for element in dictionary.values():
    print(element)
    my_writer.writerow(element)

从docs参数 towriterow应该是一个对象列表。在您的情况下，参数是一个字符串（可迭代）。所以你实际上在做的my_writer.writerow("alpha")是写成a,l,p,h,a.

你应该简单地做

my_writer.writerow(element.values())

此外，由于您的 CSV 字符串是alpha, beta, gamma. 所以当分裂发生时，元素是['aplha',' beta',' gamma',]. 你可以strip()用来删除它们

于 2013-06-30T15:13:55.567 回答
0

您正在循环遍历行中的字符而不是实际元素

with open(file_, 'r') as source, open(output, 'w') as outp:
    data = csv.reader(source, delimiter=deli)
    #line0 = next(data)
    i = 0
    for element in data:
        dictionary[i] = element
        i += 1

在这种情况下，不覆盖内置函数也被认为是一种好习惯file，而是 PEP 008 建议您在变量名称的末尾添加下划线。

于 2013-06-30T15:10:16.770 回答