pythonでcsvを取得したい

Question

pythonでcsvをリスト形式で取得し、そこから「日付」をキーにして検索し、「状況」を取得するようなものを作りたいのですが、リスト形式をどのように検索できるようにしたらよいかかわからず困っております。

csv_file.csv

日付	曜日	状況
2018/4/1	日	〇
2018/4/2	月	×
2018/4/3	火	△
…	…	…

ソースコード

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# +++ モジュール +++
import csv

# csv読み込み処理
def read_csv(self):
    # 変数
    list_data = []

    # csvファイル読み込み
    with open("./csv_file.csv", "r", encoding="ms932", errors="", newline="" ) as f:
        # csv読み込み
        reader = csv.reader(f)

        # csvのヘッダー(一行目)を読み飛ばす
        header = next(reader)

        # csvファイルをリスト形式で保存
        for row in reader:
            list_data.append(row)

        # リストデータを返す
        return list_data

# メイン処理
def main(self):
    today_is = "2018/4/2"
    # csvの読み込み
    list_data = read_csv()

    # ここから検索させたい…

# +++ 初期動作 +++
if __name__ == "__main__":
    main()

リストを辞書形式にしないとだめなのでしょうか？？
どうか教えてください。

Answer 1

そういった用途ですとpandasを利用された方が良いかと思います。

参考：
Python Data Analysis Library — pandas: Python Data Analysis Library
Pandasを使ったデータ操作の基本 - ぴよぴよ.py
10分でPandasを学ぶ - Qiita

Comments

[morutemu]

すみません、ちなみに
def read_csv(self):
# csvファイルを読み込む
gbd_df = pd.read_csv("./csv_file.csv" ,encoding='cp932')
print(gbd_df.query('日付 == "2018/4/24"'))

このような形で検索出来たのですが、状況のデータを取り出すにはどうしたら良いでしょうか？？

[hayataka2049]

このように扱えます。
>>> import pandas as pd
>>> df = pd.DataFrame([[1,2,3],[4,5,6],[7,8,9]], columns=["hoge","fuga","piyo"])
>>> df
hoge fuga piyo
0 1 2 3
1 4 5 6
2 7 8 9
>>> df.query("hoge == 1")
hoge fuga piyo
0 1 2 3
>>> df.query("hoge == 1")["piyo"]
0 3

[morutemu]

>>> df.query("hoge == 1")["piyo"]
0 3
すみません、こちらを3のみ取得したい場合はどうしたらよいでしょうか？
strなどで。。
後で、if文などに使いたいと思いまして。

[hayataka2049]

df.query("hoge == 1")["piyo"].values[0]でいけます

[morutemu]

おおおおお！ありがとうございます！
いけました！！
ありがとうございました。

[hayataka2049]

参考情報ですが、df.queryは遅いらしいので、こう書いた方が得します
df[df["hoge"] == 1]["piyo"].values[0]

参考：
https://funmatu.wordpress.com/2017/06/11/df-query%E3%82%88%E3%82%8A%E6%99%AE%E9%80%9A%E3%81%AB%E6%9B%B8%E3%81%8F%E6%96%B9%E3%81%8C%E5%A4%9A%E5%88%86%E8%89%AF%E3%81%84/

[morutemu]

おおっ！ありがとうござます！
書き換えます！

Answer 2

self不要

• def read_csv(self): などの関数定義の第1引数に self を書いていますが不要です。
• おそらく、クラスのメソッドに書かれている物を真似しようと思ったのかと思いますがこのコードの中では不要です。(今実行してもエラーになるでしょう。

実行して小さく考えてみよう

• read_csv()の結果は何になっていますか？　予定通りの結果でしょうか？
• 狙い通りになっている可能性もありますし、狙いと違うかもしれません。それを踏まえて次の事を考えましょう。

検索したい

• どうなれば検索できるたといえるのでしょうか？
• dictのキーに日付を入れておいたとして、結果はどのような形式が良いのでしょうか？

Comments

[morutemu]

そうですねぇ、イメージとしては、
raspberry piを付けっぱなしにして、毎日今日の日付を取得し、その日付からcsvの「日付」をさがして「状況」の値を取得するようなものができると嬉しいです。

Answer 3

参考情報

• pythonでのcsvファイルの読み込み

https://qiita.com/motoki1990/items/0274d8bcf1a97fe4a869