逐行向CSV文件中写入内容的方法是什么?

136

我有一些数据通过http请求进行访问,并以逗号分隔的格式由服务器发送回来,我有以下的代码:

site= 'www.example.com'
hdr = {'User-Agent': 'Mozilla/5.0'}
req = urllib2.Request(site,headers=hdr)
page = urllib2.urlopen(req)
soup = BeautifulSoup(page)
soup = soup.get_text()
text=str(soup)

文本内容如下:

april,2,5,7
may,3,5,8
june,4,7,3
july,5,6,9

我该如何将这些数据保存到CSV文件中。 我知道可以通过以下方式逐行迭代:

import StringIO
s = StringIO.StringIO(text)
for line in s:

但是我不确定如何正确地将每行写入CSV

编辑---> 感谢反馈,根据建议,解决方案相当简单,如下所示。

解决方案:

import StringIO
s = StringIO.StringIO(text)
with open('fileName.csv', 'w') as f:
    for line in s:
        f.write(line)

2
它已经是CSV格式了,你只需要将每行写入文件即可... - icedwater
2
说实话,我甚至不确定你是否需要导入StringIO。此外,当前的解决方案可能没有分隔行,因为f.write()不会自动添加换行符。 - icedwater
1
@icedwater 我明白你的意思,但我运行了上面的代码,它能够正确地将数据存储到CSV文件中。 - Mustard Tiger
1
参见:如何使用Python读写CSV文件? - Martin Thoma
6个回答

238

一般的方法:

##text=List of strings to be written to file
with open('csvfile.csv','wb') as file:
    for line in text:
        file.write(line)
        file.write('\n')

或者

使用CSV写入器:

import csv
with open(<path to output_csv>, "wb") as csv_file:
        writer = csv.writer(csv_file, delimiter=',')
        for line in data:
            writer.writerow(line)

或者

最简单的方法:

f = open('csvfile.csv','w')
f.write('hi there\n') #Give your csv text here.
## Python will convert \n to os.linesep
f.close()

25
对于 Python 3,将其更改为 with open(<输出 CSV 文件的路径>, "w", newline='') as csv_file: - Khaled Hamed
3
这行代码 for line in data: 缺少信息,请修正。谢谢。 - Francisco Maria Calisto
1
如何使用第三个解决方案将一行追加到 CSV 文件中,如果已经存在内容? - Jürgen K.
7
иҜ·дҪҝз”Ё'a'пјҲиҝҪеҠ жЁЎејҸпјүд»Јжӣҝ'w'пјҲеҶҷе…ҘжЁЎејҸпјүгҖӮ - Ani Menon
如果你遇到了这个错误:TypeError: a bytes-like object is required, not 'str',只需要将open()函数中的'wb'改为'w'即可。 - tsveti_iko

20

您可以像编写普通文件一样编写文件。

with open('csvfile.csv','wb') as file:
    for l in text:
        file.write(l)
        file.write('\n')

如果恰好是列表的列表,您可以直接使用内置的csv模块

import csv

with open("csvfile.csv", "wb") as file:
    writer = csv.writer(file)
    writer.writerows(text)

13

我会简单地将每一行写入文件,因为它已经是CSV格式:

write_file = "output.csv"
with open(write_file, "wt", encoding="utf-8") as output:
    for line in text:
        output.write(line + '\n')

我现在记不起来如何写带有换行符的行了,哈哈 :p

另外,您可能会想看一下关于 write()writelines()'\n'此答案


1
请关注编码,例如 open(write_file, 'wt', encoding='utf-8') - Bellash

9
除了之前的回答,我创建了一个用于快速写入CSV文件的类。这种方法简化了打开文件的管理和关闭,确保了一致性和更清晰的代码,特别是在处理多个文件时。
class CSVWriter():

    filename = None
    fp = None
    writer = None

    def __init__(self, filename):
        self.filename = filename
        self.fp = open(self.filename, 'w', encoding='utf8')
        self.writer = csv.writer(self.fp, delimiter=';', quotechar='"', quoting=csv.QUOTE_ALL, lineterminator='\n')

    def close(self):
        self.fp.close()

    def write(self, *args):
        self.writer.writerow(args)

    def size(self):
        return os.path.getsize(self.filename)

    def fname(self):
        return self.filename

使用示例:

mycsv = CSVWriter('/tmp/test.csv')
mycsv.write(12,'green','apples')
mycsv.write(7,'yellow','bananas')
mycsv.close()
print("Written %d bytes to %s" % (mycsv.size(), mycsv.fname()))

玩得开心


2
这是什么意思:

这是什么意思:

with open("your_csv_file.csv", "w") as f:
    f.write("\n".join(text))

返回一个字符串,该字符串是可迭代对象中所有字符串的连接结果。连接元素之间的分隔符是提供此方法的字符串。参考链接:str.join()

0

在我的情况下...

  with open('UPRN.csv', 'w', newline='') as out_file:
    writer = csv.writer(out_file)
    writer.writerow(('Name', 'UPRN','ADMIN_AREA','TOWN','STREET','NAME_NUMBER'))
    writer.writerows(lines)

你需要在 open 属性中包含 newline 选项,它就可以工作了。

https://www.programiz.com/python-programming/writing-csv-files


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接