python mongodb 设置密码前一篇ok，csv文件存入mongodb

CSV到MongoDB股票数据导入

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

原创最新推荐文章于 2024-05-01 09:45:05 发布 · 1.4k 阅读

0 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

101 篇文章

订阅专栏

本文介绍了一种使用Python将CSV文件中的股票数据批量导入MongoDB的方法。通过定义一个类来封装股票信息，并利用pymongo库实现数据写入。示例中展示了如何读取CSV文件并逐条插入数据，最后统计集合中的文档数量。

部署运行你感兴趣的模型镜像

# coding:utf-8
import os
import csv
import pymongo
import time
'''
mongodb 删除数据库
use test;
db.dropDatabase();
mongodb删除表
db.mytable.drop();
清空表
db.mytable.remove({})
'''
class c_stock:
    def __init__(self, code, date, open, high, low, close,volume, adj_close):
        self.code = code
        self.date = date
        self.open = open
        self.high = high
        self.low = low
        self.close = close
        self.volume = volume
        self.adj_close = adj_close

#mongodb_link = 'mongodb://127.0.0.1:27017'
#mongoClient = MongoClient(mongodb_link)
conn = pymongo.MongoClient("localhost", 27017)
conn.db_stock.authenticate("d","zz")
db = conn.db_stock
stocks = db.stocks

def write_dict(type):
    db.stocks.insert(type.__dict__)

'''
with open('000002.csv') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        one = c_stock(code='000002',
                     date=row['date'],
                     open=float(row['open']),
                     high=float(row['high']),
                     low=float(row['low']),
                     close=float(row['close']),
                     volume=int(row['volume']),
                     adj_close=float(row['adj_close']),
                     )
        write_dict(one)
'''
#db.stocks.drop()
#db.stocks.remove({})
reader = csv.reader(file('000002.csv','rb'))
for row in reader:
    if reader.line_num == 1:
        continue
    one = c_stock(code='000002',
                      date=row[0],
                      open=float(row[1]),
                      high=float(row[2]),
                      low=float(row[3]),
                      close=float(row[4]),
                      volume=int(row[5]),
                      adj_close=float(row[6]))

    write_dict(one)

print db.stocks.count()
#time.sleep(3)
#for i in db.stocks.find():
#    print i