python - 用Python编写csv文件: unwanted commas appear

Question

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

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 输入输出

但是，尝试使用 csv 文件，例如:阿尔法、贝塔、 Gamma 、德尔塔

我得到以下结果:

a,l,p,h,a

 ,b,e,t,a

 ,g,a,m,m,a

 ,d,e,l,t,a

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

我错过了什么吗？

Answer 1

问题出在行

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

来自 docs writerow 的参数应该是对象列表。在你的例子中，参数是一个字符串(它是可迭代的)。所以你实际上在做的是 my_writer.writerow("alpha") ，它被写成 a,l,p,h,a 。

你应该简单地这样做

my_writer.writerow(element.values())

此外，您还会得到前导逗号，因为您的 CSV 字符串是 alpha, beta, gamma。因此，当分割发生时，元素为['aplha','beta','gamma',]。您可以使用 strip() 删除它们