网络编程
位置：首页>> 网络编程>> Python编程>> Python访问MongoDB,并且转换成Dataframe的方法

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

作者：宝山方圆　　发布时间：2022-07-18 20:32:07　

标签：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")

小结：

1. 解决了pymongo访问MongoDB的问题

2. 解决了查询的数据转成dataframe的问题

3. 解决了dataframe写入到csv的问题

4. 解决了中文乱码问题。

来源：https://www.cnblogs.com/zhzhang/p/9305920.html

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com