方法一: 使用pandas 写入csv文件
import pandas as pd
a = ["a", "b", "c"]
b = ["d", "e", "f"]
dataframe = pd.DataFrame({'one_name': a, 'two_name': b})
dataframe.to_csv("test.csv", index=False, sep=',')
方法二:使用csv库一行一行的写入csv文件
import csv
with open("test.csv","w", 'utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(["index","a_name","b_name"])
writer.writerows([[0,1,3],[1,2,3],[2,3,4]])
示例:使用pymongo读取mongo数据库内容并写入csv文件
import codecs
import csv
import pymongo
mongo_client = pymongo.MongoClient('127.0.0.1', 27017)
db = mongo_client.ceis_nlp
my_collection = db.twitter_tweet
cursor = my_collection.find()
with codecs.open('data.csv', 'w', 'utf-8') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(["name", "content", "topic", "at"])
for data in cursor:
writer.writerows([[data["name"], data["content"], data["topic"], data["at"]]])