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")