pythonでCSVファイルを、ユーザ入力で行・列指定について

Question

#実現したいこと
CSVファイルを読み込んだ後、B列とC列をユーザ入力で日付と時間を指定してX軸に、ユーザ入力でD列とE列に適切な値を入力してJ列以降のデータ列をY軸に当て、最後は以上の条件に当てはまったX軸,Y軸のグラフを出力させたい。
#問題点
input()関数を使用してユーザ入力を試みましたが、入力した文字列・値とCSVファイルの照合が出来ませんでした。グラフの行列ラベルとユーザ入力で指定した文字列・数値を一致させる方法がわかりませんので、ご教授をお願い致します。

#使用環境
Python3.X
Jupyter Notebook

Answer 1

まず時間をまとめて、datetimeに変換します。

入力した値をdatetimeオブジェクトとして処理できれば、後はマスクを作って抽出することができます。

import pandas as pd
import datetime

N = 10000
base = datetime.datetime.today().replace(microsecond=0)
date_list = [base - datetime.timedelta(hours=x) for x in range(0, N*5, 5)][::-1]
date_date = [str(d.date()) for d in date_list]
date_time = [str(d.time()) for d in date_list]
df = pd.DataFrame(columns=['Date', 'ClockTime', 'target'])
df['Date'] = date_date
df['ClockTime'] = date_time
df['target'] = np.random.rand(N)
print(df.head())
'''
         Date ClockTime    target
0  2012-03-09  13:22:15  0.525087
1  2012-03-09  18:22:15  0.242866
2  2012-03-09  23:22:15  0.720774
3  2012-03-10  04:22:15  0.168702
4  2012-03-10  09:22:15  0.216134
'''

tdf = df['Date'] + ' ' + df['ClockTime']
tdf = pd.to_datetime(tdf)
df['DateTime'] = tdf
print(df.head())
'''
         Date ClockTime    target            DateTime
0  2012-03-09  13:22:15  0.525087 2012-03-09 13:22:15
1  2012-03-09  18:22:15  0.242866 2012-03-09 18:22:15
2  2012-03-09  23:22:15  0.720774 2012-03-09 23:22:15
3  2012-03-10  04:22:15  0.168702 2012-03-10 04:22:15
4  2012-03-10  09:22:15  0.216134 2012-03-10 09:22:15
'''

mask = ((datetime.datetime.now() - 
         datetime.timedelta(days=10))< tdf) & (tdf < datetime.datetime.now())
print(df[mask].head())
'''
            Date ClockTime    target            DateTime
9952  2017-11-11  21:22:15  0.523548 2017-11-11 21:22:15
9953  2017-11-12  02:22:15  0.797870 2017-11-12 02:22:15
9954  2017-11-12  07:22:15  0.777698 2017-11-12 07:22:15
9955  2017-11-12  12:22:15  0.020755 2017-11-12 12:22:15
9956  2017-11-12  17:22:15  0.390619 2017-11-12 17:22:15
'''
print(df[mask].tail())
'''
            Date ClockTime    target            DateTime
9995  2017-11-20  20:23:27  0.419750 2017-11-20 20:23:27
9996  2017-11-21  01:23:27  0.738802 2017-11-21 01:23:27
9997  2017-11-21  06:23:27  0.113842 2017-11-21 06:23:27
9998  2017-11-21  11:23:27  0.025397 2017-11-21 11:23:27
9999  2017-11-21  16:23:27  0.361001 2017-11-21 16:23:27
'''

Comments

[bkts94441]

ご回答ありがとうございます。記述されたプログラム、ヒントを参考にプログラムを作成しようと思います。

Answer 2

df.locにて複数の列の値をand条件にて行を抽出する例です。
行の絞り条件をユーザーが入力するものと解釈しました。

なお、Date,ClockTimeについては日時として扱った方が範囲指定など利用しやすいため、その例も示しています。

import numpy as np
import pandas as pd
from io import StringIO

s = """
xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx
0    xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx
1    xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx
2    xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx
3    xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx
4    xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx
5    xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx   xxxx
"""
df = pd.read_csv( StringIO(s), sep=" ", skipinitialspace=True)
#print(df)

# Date,ClockTimeは文字列、motion_type,speed_settingは数値として一致判定して行を抽出
inp_Date = '2017/8/24'
inp_ClockTime = '00:00.0'
inp_mt = 0
inp_ss = 0

df_sel = df.loc[(df['Date']==inp_Date) & (df['ClockTime']==inp_ClockTime) & (df['motion_type']==inp_mt) & (df['speed_setting']==inp_ss)]
print(df_sel[['Nom','Date','ClockTime','motion_type','speed_setting']])

# 日付と時刻を'DateTime'列にまとめ、日時の範囲で抽出

df['DateTime'] = pd.to_datetime( df['Date'] + ' ' + df['ClockTime'])

inp_date_st = pd.to_datetime('2017/8/24 01:00:00')
inp_date_ed = pd.to_datetime('2017/8/25 00:00:00')

df_sel = df.loc[(df['DateTime'] >= inp_date_st) & (df['DateTime'] < inp_date_ed)]
print(df_sel[['Nom','Date','ClockTime']])

Comments

[bkts94441]

ご回答ありがとうございます。記述されたプログラムを元に作成してみようと思います。