複数のcsvファイルをPythonでMongoDBにインポートするやり方がわからない。

1日ごとに分けられたcsvファイルが1ヶ月分ありcsvファイルの中は1分ごとに行が区切られそれが24時間あります。
そのcsvファイルをMongoDBで管理しようと思っているのですがPythonでインポートするやり方がわかりません。

以下はMongoDBにドキュメントを追加するコードを試したものです。
このコードはうまくいったのですがcsvファイルをインポートするやり方がわかりません。

Python
1from pymongo import MongoClient
2from datetime import datetime
3
4class TestMongo(object):
5
6     def __init__(self):
7         self.clint = MongoClient()
8         self.db = self.clint['test']
9
10     def add_one(self):
11        """データ挿入"""
12        post = {
13            'title': 'ハリネズミ',
14            'content': 'ハリネズミ可愛い~',
15            'created_at': datetime.now()
16        }
17        return self.db.test1.insert_one(post)
18
19def main():
20    obj = TestMongo()
21    rest = obj.add_one()
22    print(rest)
23
24if __name__ == '__main__':
25    main()

よろしくお願いします。

TakaiY

2020/11/17 09:19

試したコードはどうだったのでしょう？どんな問題が発生したのですか？

Lapis051130

2020/11/18 01:43 編集

試したコードはうまくいきました。 csvファイルをインポートするやり方がわからないのです。

行動規範の内容に同意します

回答2件

ベストアンサー

各行を以下のようにdict形式で取り出すことができれば、MongoDBへの投入はできるということですよね

python
1        post = {
2            'title': 'ハリネズミ',
3            'content': 'ハリネズミ可愛い~',
4            'created_at': datetime.now()
5        }

csvモジュールを使うとCSVファイルの各行をdict形式で取得できます。

python
1import csv
2
3with open('test.csv', 'r', encoding='utf-8') as f:
4    reader = csv.DictReader(f)
5    for l in reader:
6        print(l)

このときcsvは先頭行がカラム名になったものを使っています。

csv
1title,content,created_at
2ハリネズミ,ハリネズミ可愛い,2020-10-10T10:20:30
3針山,針山硬い,2020-11-11T01:02:03

動作結果は、

text
1import csv...
2OrderedDict([('title', 'ハリネズミ'), ('content', 'ハリネズミ可愛い'), ('created_at', '2020-10-10T10:20:30')])
3OrderedDict([('title', '針山'), ('content', '針山硬い'), ('created_at', '2020-11-11T01:02:03')])

こんな感じです。
型がOrderdDictになってますが、dictと同じように使えるはずなので、printしているタイミングでDBに投入すればいいはず。

投稿2020/11/18 03:12

TakaiY

総合スコア13790