Python访问MongoDB,并且转换成Dataframe的方法_Python

Python访问MongoDB,并且转换成Dataframe的方法

2021-04-07 00:32宝山方圆 Python

今天小编就为大家分享一篇Python访问MongoDB,并且转换成Dataframe的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

如下所示：

									#!/usr/bin/env python

									# -*- coding: utf-8 -*-

									# @Time : 2018/7/13 11:10

									# @Author : baoshan

									# @Site : 

									# @File : pandans_pymongo.py

									# @Software: PyCharm Community Edition

									import pymongo

									import pandas as pd

									def _connect_mongo(host, port, username, password, db):

									 """ A util for making a connection to mongo. """

									 if username and password:

									  mongo_uri = "mongodb://%s:%s@%s:%s/%s" % (username, password, host, port, db)

									  conn = pymongo.MongoClient(mongo_uri)

									 else:

									  conn = pymongo.MongoClient(host, port)

									 return conn[db]

									def read_mongo(db, collection, query={}, host='test43', port=27017, username=None, password=None, no_id=True):

									 """ Read from Mongo and Store into DataFrame. """

									 # Connect to MongoDB

									 db = _connect_mongo(host=host, port=port, username=username, password=password, db=db)

									 # Make a query to the specific DB and Collection

									 cursor = db[collection].find(query).limit(10)

									 # Expand the cursor and construct the DataFrame

									 df = pd.DataFrame(list(cursor))

									 df.to_csv("abc.csv", encoding="utf_8_sig") # 处理中文乱码问题

									 if no_id:

									  del df['_id']

									 return df

									read_mongo(db='service', collection='trace_log_regular', query={}, host='xxx', port=27017, username="xxx", password="xxx")