使用Python读取mongoDB写入csv文件

最新推荐文章于 2024-05-06 19:45:51 发布

原创最新推荐文章于 2024-05-06 19:45:51 发布 · 4.6k 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#csv #python #pandas #mongo

python学习笔记专栏收录该内容

29 篇文章

订阅专栏

本文介绍了两种使用Python写入CSV文件的方法：一是利用pandas库快速生成并保存数据；二是通过csv库逐行写入数据。同时，展示了如何从MongoDB数据库读取数据并将其转换为CSV格式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

方法一: 使用pandas 写入csv文件

import pandas as pd
#任意的多组列表
a = ["a", "b", "c"]
b = ["d", "e", "f"]
#字典中的key值即为csv中列名
dataframe = pd.DataFrame({'one_name': a, 'two_name': b})
#将DataFrame存储为csv,index表示是否显示行名，default=True
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)

    #先写入columns_name
    writer.writerow(["index","a_name","b_name"])
    #写入多行用writerows 每一个列表为一行
    writer.writerows([[0,1,3],[1,2,3],[2,3,4]])

示例:使用pymongo读取mongo数据库内容并写入csv文件

import codecs
import csv
import pymongo
# 链接mongo数据库
mongo_client = pymongo.MongoClient('127.0.0.1', 27017)
db = mongo_client.ceis_nlp
my_collection = db.twitter_tweet
#查询库中数据
cursor = my_collection.find()
#打开csv文件
with codecs.open('data.csv', 'w', 'utf-8') as csvfile:
    writer = csv.writer(csvfile)
    # 先写入columns_name
    writer.writerow(["name", "content", "topic", "at"])
    #写入多行用writerows
    for data in cursor:
        writer.writerows([[data["name"], data["content"], data["topic"], data["at"]]])