用 Python 将数据写到 CSV 文件

Python 标准库中，有个名为 csv 的库，专门处理 csv 的读写操作。具体使用实例如下：

#!/usr/bin/env python
#-*- coding:utf-8 -*-

import csv
import codecs
# codecs 是自然语言编码转换模块

fileName = 'test.csv'

# 指定编码为 utf-8, 避免写 csv 文件出现中文乱码
with codecs.open(fileName, 'w', 'utf-8') as csvfile:
    # 指定 csv 文件的头部显示项
    filednames = ['书名', '作者','出版社']
    writer = csv.DictWriter(csvfile, fieldnames=filednames)

    books = []
    book = {
        'title': '笑傲江湖',
        'author': '金庸',
        'publish':'人民出版社'
    }
    books.append(book)

    writer.writeheader()
    for book in books:
        try:
            writer.writerow({'书名':book['title'], '作者':book['author'],'出版社':book['publish']})
        except UnicodeEncodeError:
            print("编码错误, 该数据无法写到文件中, 直接忽略该数据")

这种方式是逐行往 CSV 文件中写数据， 所以效率会比较低。如果想批量将数据写到 CSV 文件中，需要用到 pandas 库。

pandas 是第三方库，所以使用之前需要安装。通过 pip 方式安装是最简单、最方便的。

pip install pandas

使用 pandas 批量写数据的用法如下：

#!/usr/bin/env python
#-*- coding:utf-8 -*-

import pandas as pd
fileName = 'test.csv'
number = 1

books = []
book = {
    'title': '笑傲江湖',
    'author': '金庸',
}
# 如果 book 条数足够多的话，pandas 会每次往文件中写 50 条数据。
books.append(book)

data = pd.DataFrame(books)
# 写入csv文件,'a+'是追加模式
try:
    if number == 1:
        csv_headers = ['书名', '作者']
        data.to_csv(fileName, header=csv_headers, index=False, mode='a+', encoding='utf-8')
    else:
        data.to_csv(fileName, header=False, index=False, mode='a+', encoding='utf-8')
        number = number + 1
except UnicodeEncodeError:
    print("编码错误, 该数据无法写到文件中, 直接忽略该数据")