【问题标题】:Writing a Python list of lists to a csv file将 Python 列表写入 csv 文件
【发布时间】:2012-12-11 19:57:42
【问题描述】:

我有一长串如下形式的列表---

a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]

即列表中的值具有不同的类型——浮点数、整数、字符串。如何将其写入 csv 文件,以便我的输出 csv 文件看起来像

1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2

【问题讨论】:

    标签: python file csv file-io python-2.7


    【解决方案1】:

    Python 内置的CSV module 可以轻松处理这个问题:

    import csv
    
    with open("output.csv", "wb") as f:
        writer = csv.writer(f)
        writer.writerows(a)
    

    这假设您的列表被定义为a,就像您的问题一样。您可以通过各种可选参数将输出 CSV 的确切格式调整为 csv.writer(),如上面链接的库参考页面中所述。

    Python 3 更新

    import csv
    
    with open("out.csv", "w", newline="") as f:
        writer = csv.writer(f)
        writer.writerows(a)
    

    【讨论】:

    • 为了兼容 Python 3,请从“wb”中删除“b”。
    • 使用 Python 3 - open('output.csv', 'w', newline='')。如果我省略换行参数,我会得到一个额外的行。 docs.python.org/3/library/csv.html#csv.writer
    • 在 python3 中我不得不使用 open('output.csv', 'w', newline="")
    • 哇,python 3 错误非常无用。谢谢@vladV(需要一个类似字节的对象,而不是'str')。事后看来,这有点道理,但根本不知道在哪里看。
    • @tlalco 这可能意味着您使用的是 Python 2,在这种情况下,您应该使用第一个代码块而不是第二个代码块。 (这也意味着您应该考虑切换到 Python 3。)
    【解决方案2】:

    你可以使用pandas:

    In [1]: import pandas as pd
    
    In [2]: a = [[1.2,'abc',3],[1.2,'werew',4],[1.4,'qew',2]]
    
    In [3]: my_df = pd.DataFrame(a)
    
    In [4]: my_df.to_csv('my_csv.csv', index=False, header=False)
    

    【讨论】:

    • 如果内置库csv可以做到,我认为不应该使用pandas
    • 我喜欢熊猫,因为它很强大
    • pandas 确实很强大,但我不会开迈凯轮车去隔壁的街角商店。
    • 注意:对于列表非常大的情况,顺序 csv 写入操作会使代码显着变慢。在这种情况下,使用 pandas 会更明智。
    【解决方案3】:
    import csv
    with open(file_path, 'a') as outcsv:   
        #configure writer to write standard csv file
        writer = csv.writer(outcsv, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL, lineterminator='\n')
        writer.writerow(['number', 'text', 'number'])
        for item in list:
            #Write item to outcsv
            writer.writerow([item[0], item[1], item[2]])
    

    官方文档:http://docs.python.org/2/library/csv.html

    【讨论】:

    • 如果你能用一些 cmets 解释你的答案,这将得到我的 +1。
    • writerow 不接受多个参数。
    • >>> w.writerow("a", "b", "c") Traceback(最近一次调用最后):文件“”,第 1 行,在 TypeError : writerow() 只接受一个参数(给定 3 个)
    • @Amber 你用的是什么版本的python?
    • @Amber 对不起。我错过了[]
    【解决方案4】:

    在我非常大的列表中使用 csv.writer 需要很长时间。我决定使用 pandas,它更快、更容易控制和理解:

     import pandas
    
     yourlist = [[...],...,[...]]
     pd = pandas.DataFrame(yourlist)
     pd.to_csv("mylist.csv")
    

    你可以改变一些东西来制作一个更好的 csv 文件:

     yourlist = [[...],...,[...]]
     columns = ["abcd","bcde","cdef"] #a csv with 3 columns
     index = [i[0] for i in yourlist] #first element of every list in yourlist
     not_index_list = [i[1:] for i in yourlist]
     pd = pandas.DataFrame(not_index_list, columns = columns, index = index)
    
     #Now you have a csv with columns and index:
     pd.to_csv("mylist.csv")
    

    【讨论】:

      【解决方案5】:

      如果出于某种原因您想手动执行此操作(不使用 csvpandasnumpy 等模块):

      with open('myfile.csv','w') as f:
          for sublist in mylist:
              for item in sublist:
                  f.write(item + ',')
              f.write('\n')
      

      当然,滚动您自己的版本可能容易出错且效率低下......这通常是为什么有一个模块来解决这个问题。但有时编写自己的代码可以帮助您了解它们的工作原理,有时甚至更容易。

      【讨论】:

        【解决方案6】:

        如果您不想为此导入 csv 模块,您可以仅使用 Python 内置函数将列表列表写入 csv 文件

        with open("output.csv", "w") as f:
            for row in a:
                f.write("%s\n" % ','.join(str(col) for col in row))
        

        【讨论】:

          【解决方案7】:

          Ambers 的解决方案也适用于 numpy 数组:

          from pylab import *
          import csv
          
          array_=arange(0,10,1)
          list_=[array_,array_*2,array_*3]
          with open("output.csv", "wb") as f:
              writer = csv.writer(f)
              writer.writerows(list_)
          

          【讨论】:

            【解决方案8】:

            将列表列表转储到pickle并使用pickle module恢复它怎么样?这很方便。

            >>> import pickle
            >>> 
            >>> mylist = [1, 'foo', 'bar', {1, 2, 3}, [ [1,4,2,6], [3,6,0,10]]]
            >>> with open('mylist', 'wb') as f:
            ...     pickle.dump(mylist, f) 
            
            
            >>> with open('mylist', 'rb') as f:
            ...      mylist = pickle.load(f)
            >>> mylist
            [1, 'foo', 'bar', {1, 2, 3}, [[1, 4, 2, 6], [3, 6, 0, 10]]]
            >>> 
            

            【讨论】:

              【解决方案9】:

              创建作者时一定要注明lineterinator='\n';否则,当数据源来自其他 csv 文件时,可能会在每个数据行之后将额外的空行写入文件...

              这是我的解决方案:

              with open('csvfile', 'a') as csvfile:
                  spamwriter = csv.writer(csvfile, delimiter='    ',quotechar='|', quoting=csv.QUOTE_MINIMAL, lineterminator='\n')
              for i in range(0, len(data)):
                  spamwriter.writerow(data[i])
              

              【讨论】:

                【解决方案10】:

                如果将 listslistslll list 导出到 .csv,这将在 Python3 中工作:

                import csv
                with open("output.csv", "w") as f:
                    writer = csv.writer(f)
                    for element in lll:
                        writer.writerows(element)
                

                【讨论】:

                  【解决方案11】:

                  在 csv.writer 函数中使用 newline 参数执行示例时,我收到一条错误消息。以下代码对我有用。

                   with open(strFileName, "w") as f:
                      writer = csv.writer(f, delimiter=',',  quoting=csv.QUOTE_MINIMAL)
                      writer.writerows(result)
                  

                  【讨论】:

                    【解决方案12】:

                    在此页面上什至没有看到一个答案,其中包括如何包含标题以及创建文件。这是一种结合了它的方法。方法适用于 python 3

                    csv_filename = 'abc.csv'
                    fieldnames = ['Col_1','Col_2','Col_3','Col_4']
                    with open(csv_filename, mode='w') as csv_file:
                        writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
                        writer.writeheader()
                        for i in my_list:
                            writer.writerow({fieldnames[0]: i[0], fieldnames[1]: i[1], fieldnames[2]: i[2],fieldnames[3]: i[3]}) 
                    

                    【讨论】:

                      猜你喜欢
                      • 1970-01-01
                      • 1970-01-01
                      • 2014-06-27
                      • 2016-09-02
                      • 2016-03-26
                      • 1970-01-01
                      • 1970-01-01
                      相关资源
                      最近更新 更多