前提
Pythonでラグ特徴量を作成しようとしているのですが、日にちのインデックスが重複しており日時の重複分を加味した行のスライドができません。
上手くシフトさせる方法を教えて欲しいです。
実現したいこと
・日付(=date)重複分(重複データの数はバラバラ)を加味して日付もしくは労働者数(=worker)の値が変化した際に新しいDataframeに対して次の値に前の値を入れたいです。
・日付ごとに件数は変わります。また、日付と労働者数は同じ動きをします。
#元の状態 date worker 0 20180101 1000 1 20180101 1000 2 20180103 2000 3 20180103 2000 4 20180103 2000 5 20180104 500 #実現したいこと date worker worker_lag 0 20180101 1000 NaN 1 20180101 1000 NaN 2 20180103 2000 1000 3 20180103 2000 1000 4 20180103 2000 1000 5 20180104 500 2000
該当のソースコード
Python
1import pandas as pd 2df = pd.DataFrame( 3 data={'date': ['20180101', '20180101', '20180103', '20180103','20180103','20180104'], 4 'worker': [1000, 1000, 2000, 2000, 2000, 500]} 5) 6 7df['worker_lag'] = df.groupby(['date'])["worker"].shift() 8 9### 試したこと 10df['worker_lag'] = df.groupby(['date','worker'])['worker'].shift() 11#1行分しかずれませんでした。
回答2件
あなたの回答
tips
プレビュー
下記のような回答は推奨されていません。
このような回答には修正を依頼しましょう。
2022/11/08 10:05