PYTHONのPANDASを利用し分ごとの売買の合計と累積を出したい

Question

PYTHONのPANDASを利用して、
分ごとの売り買いの合計
前の1分との累積
を算出したいのですが、考え方もわからず困っております。。。。どなたか助けて頂けないでしょうか。

データは以下のような形で　時間（マイクロ秒まで）と数量（小数点あり）　売買（１が買い　２が売り）　です。

時間　　 数量 売買
15:00:05 1.0 1
15:00:08 1.0 2
15:01:12 2.0 1
15:02:15 2.0 2

このデータを

時間 数量 累積
15:00 0.0　0
15:01 2.0　2
15:02 -2.0　0
このような形にしたいです。

分以下を切り捨てして、　売りならマイナスの数量にするイメージです

Comments

[meg_]

時間のデータタイプは何でしょうか？(15:01:12の15は分ですか？)

[shinnzinbuyer]

返事ができていませんでした。
失礼しました。
データタイプはdatetimeでマイクロ秒が入っているものでした。

Answer 1

順番に処理するとこんな感じになります

import pandas as pd
import io

data = """
時間,数量,売買
15:00:05,1.0,1
15:00:08,1.0,2
15:01:12,2.0,1
15:02:15,2.0,2
"""

# データ読み込み(時間列をDatetimeIndexに指定)
df = pd.read_csv(io.StringIO(data), parse_dates=['時間'], index_col='時間')
# 数量列に売買列に応じた符号を追加
df['数量'] = df['売買'].map({1:1,2:-1}) * df['数量']
# 売買列は不要なので削除
df = df.drop('売買', axis=1)
# 分毎にリサンプル
df = df.resample('1min').agg({'数量':'sum'})
# 累計列を追加
df['累計'] = df['数量'].cumsum()

print(df)
#                      数量   累計
#時間
#2019-10-11 15:00:00  0.0  0.0
#2019-10-11 15:01:00  2.0  2.0
#2019-10-11 15:02:00 -2.0  0.0

Comments

[shinnzinbuyer]

ありがとうございます、早速試してみます！