Question

我正在编写一个脚本来帮助我操作.csv文件。目标是读取指定的.csv，然后将数据拆分为几个临时文件以供进一步操作。输入.csv中的空行（空字符串列表）表示我想要分割数据的位置。

如果我的代码针对PEP 8中的任何内容运行，我深表歉意。对于我来说，Python（以及一般的编码）对我来说仍然是非常新的。

import os
import csv
import tempfile

def importFromCSV(filepath):
    print("Reading data from",os.path.basename(filepath))
    datalist = []
    with open(filepath) as csvfile:
        file_dialect = csv.Sniffer().sniff(csvfile.readline(),[',',';',':','\t','.'])
        csvfile.seek(0)
        filereader = csv.reader(csvfile,dialect = file_dialect)
        for row in filereader:
            datalist.append(row)
    return datalist

def SplitToTemp(datalist, target_dir):
    tmpnamelist = []
    templist = []
    for item in datalist:
        if item[0] != '':
            templist.append(item)
        else:
            del item
            f = tempfile.NamedTemporaryFile(delete = False, dir = target_dir)
            tmpnamelist.append(f.name)
            dw = csv.writer(f, delimiter = ',', quotechar = '|', quoting = csv.QUOTE_MINIMAL)
            for row in templist:
                dw.writerow(row)
            f.close()
            templist = []
    return tmpnamelist

###############################################################################
pathname = os.path.normpath('C:/Python33/myprograms/myclassandfx/BenchLink/blrtest.csv')
tempdir = tempfile.mkdtemp(dir = os.path.normpath('c:/users/'+os.getlogin()+'/desktop'))

filedata = import_from_csv(pathname)
tempnames = SplitToTemp(filedata, tempdir)

然而，当我运行代码时，我遇到了这个：

Traceback (most recent call last):
  File "C:\Python33\myprograms\myclassandfx\BenchLink\BenchLinkReader_classless.py", line 56, in <module>
    tempnames = SplitToTemp(filedata, tempdir)
  File "C:\Python33\myprograms\myclassandfx\BenchLink\BenchLinkReader_classless.py", line 45, in     SplitToTemp
    dw.writerow(row)
TypeError: 'str' does not support the buffer interface

困扰我的部分是，当我print(temp)时，我仍然会得到一份清单。

我在这里做错了什么？

Answer 1

默认情况下，tempfile.NamedTemporaryFile()对象以二进制模式打开;引自documentation：

模式参数默认为'w+b'，因此可以在不关闭的情况下读取和写入创建的文件。使用二进制模式，使其在所有平台上的行为一致，而不考虑存储的数据。

您需要以文本模式打开它：

f = tempfile.NamedTemporaryFile(mode='w+', delete=False, dir=target_dir)

当dw.writerow()方法尝试将unicode str值写入仅支持bytes的文件对象（支持缓冲区接口的对象）时，抛出错误。

使用csv.writer遇到TypeError

1 个答案: