9

我正在尝试将列表列表写入新文件，但出现此错误：

回溯（最后一次调用）：文件“”，第 1 行，在 dowork() 文件“C:\Python27\work\accounting\formatting quickbooks file\sdf.py”，第 11 行，在 dowork WriteFile() 文件“C :\Python27\work\accounting\formatting quickbooks file\sdf.py", line 71, in WriteFile f.write(thefile) TypeError: expected a character buffer object

如何将列表的列表写入文件？

这就是我的写作方式：

def WriteFile():
    global thefile
    f = open("output"+'.csv','w')
    f.seek(0)
    f.write(thefile)
    f.close()

如果您需要，这里是完整的源代码：

import csv

thefile = []
output = []

def dowork():
    sourceFile='e.csv'
    thefile=ReadFile(sourceFile)
    CleanFile(sourceFile)
    ProcessFile()
    WriteFile()

def ReadFile(filename):
    return list(csv.reader(open(filename, 'rb'), delimiter=',', quotechar='"'))[1:]

def CleanFile(sourceFile):
    global thefile
    thefiletmp=[]
    for i, line in enumerate(thefile):
        if line[2]=='':
            del thefile[i]
        else:
            thefiletmp.append(line[4:])
    thefile=thefiletmp


def ProcessFile():
    global thefile
    iCompany=1
    iNum=0
    iDate=2
    iAging=3
    iBalance=4
    COMPANIES=GetDistinctValues(1)
    mytemparray=[]
    mytempfile=[]
    for company in COMPANIES:
        for line in thefile:
            if line[iCompany]==company:
                mytemparray.append(line[iCompany])
                mytemparray.append(line[iNum])
                mytemparray.append(line[iDate])
                if line[2] in range(0,31):
                    mytemparray.append(line[iBalance])
                    mytemparray.append('0')
                    mytemparray.append('0')
                    mytemparray.append('0')
                if line[2] in range(31,61):
                    mytemparray.append('0')
                    mytemparray.append(line[iBalance])
                    mytemparray.append('0')
                    mytemparray.append('0')
                if line[2] in range(61,91):
                    mytemparray.append('0')
                    mytemparray.append('0')
                    mytemparray.append(line[iBalance])
                    mytemparray.append('0')
                if line[2] >90:
                    mytemparray.append('0')
                    mytemparray.append('0')
                    mytemparray.append('0')
                    mytemparray.append(line[iBalance])
                mytempfile.append(mytemparray)
                mytemparray=[]
    thefile=mytempfile

def WriteFile():
    global thefile
    f = open("output"+'.csv','w')
    f.seek(0)
    f.write(thefile)
    f.close()

def GetDistinctValues(theColumn):
    return sorted(list(set(line[theColumn] for line in thefile)))
4

3 回答 3

15

错误消息的意思是您不能将列表写入文件，只能将“字符缓冲区对象”写入文件，这意味着字符串或其他类似于字符串的东西。

如果您只想以与将列表打印到控制台相同的方式将列表写入文件，则可以编写str(thefile)or repr(thefile)（甚至使用重定向语法print而不是 using file.write）。

但是您正在使用该csv模块来读取输入，并且可能希望输出格式相同，因此您可能也想使用csv它来编写它。

你是这样读的：

list(csv.reader(open(filename, 'rb'), delimiter=',', quotechar='"'))[1:]

所以这样写：

csv.writer(open('foo.csv', 'wb'), delimiter=',', quotechar='"').writerows(thefile)

我应该提一下，我一开始不会像这样构造代码；我会做这样的事情：

with open('input.csv', 'rb') as infile, open('output.csv', 'wb') as outfile:
  incsv = csv.reader(infile, delimiter=',', quotechar='"')
  outcsv = csv.writer(outfile, delimiter=',', quotechar='"')
  incsv.read() # skip first line
  for line in incsv:
    if line[3] != '':
      outcsv.write(ProcessLine(line))
于 2012-09-26T00:10:28.690 回答
4

您不能将列表写入文件；你只能写一个字符串。以某种方式将列表转换为字符串，或者遍历列表并一次写入一个元素。或者您正在使用该csv模块，使用它来写入文件。

顺便说一句，调用文件以外的东西（例如列表）thefile必然会导致混乱。

于 2012-09-25T23:28:46.090 回答
2

thefile 是列表列表，而不是字符缓冲区。

for sublist in thefile:
    f.write("".join(sublist))  # perhaps

这里有一些不好的地方，使用全局，命名文件列表，...

（正确答案是 abarnert 的）。

于 2012-09-25T23:29:00.090 回答