利用pandas读取MongoDB库中的数据

最新推荐文章于 2024-05-01 09:45:05 发布

原创

最新推荐文章于 2024-05-01 09:45:05 发布 · 719 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#pandas #mongodb #数据库

下方代码的主要目的是从MongoDB数据库中获取数据，并使用pandas库将其转换为DataFrame。

# coding=utf-8
from pymongo import MongoClient
import pandas as pd

# 创建MongoDB客户端连接
client = MongoClient()
# 选择数据库'douban'中的集合'tv1'
collection = client['douban']['tv1']
# 从集合中查询所有数据
data = collection.find()

# 初始化一个空列表，用于存储处理后的数据
data_list = []

# 遍历查询结果
for i in data:
    # 创建一个临时字典，用于存储每条记录的特定字段
    temp = {}
    temp['info'] = i['info']
    temp['rating_count'] = i['rating']['count']
    temp['rating_value'] = i['rating']['value']
    temp['title'] = i['title']
    temp['country'] = i['tv_category']
    temp['directors'] = i['directors']
    temp['actors'] = i['actors']
    # 将临时字典添加到列表中
    data_list.append(temp)

# 设置pandas的显示选项，以便更好地查看DataFrame
pd.set_option('display.max_columns', 1000)
pd.set_option('display.width', 1000)
pd.set_option('display.max_colwidth', 1000)

# 将处理后的数据列表转换为DataFrame
df = pd.DataFrame(data_list)

# 打印DataFrame的前1行
print(df.head(1))

# 打印DataFrame的'info'列，并将其字符串