#実現したいこと
列名の一部とそれに付随する数字等を組み合わせて、新しい列名と計算結果をデータフレームに追加したいと考えています。
現時点において、列名の変数展開ができておらず、そもそもできないのか、書き方が悪いのかわかりません。
特に変数展開の部分においてだけでもご教示いただけると大変助かります。
#ループを使わない場合
以下のコードを実行することで、欲しい結果は得られます。
しかし,元データの売上高以外の項目も同様に成長率等を求めたいと考えていますが、同じことを書くのは無駄なように感じています。なお、この場合は、変数展開の問題はありません。
Python
1def make_indicator_multiple(): 2 df = pd.read_csv(Path,encoding="cp932") 3 #売上高分析 4 df['n年前'] = df.groupby('証券コード')['年度'].transform('max')-df['年度'] 5 6 df=df.pivot(index='証券コード',columns='n年前',values='売上高') 7 df['売上高成長率_1→0'] = (df[0]-df[1])/df[1] 8 df['売上高成長率_2→1'] = (df[1]-df[2])/df[2] 9 df['売上高成長率_3→2'] = (df[2]-df[3])/df[3] 10 df['売上高成長率_4→3'] = (df[3]-df[4])/df[4] 11 df['売上高CAGR_2→0'] = (df[0]/df[3])**(1/(2-0))-1 12 df['売上高CAGR_4→0'] = (df[0]/df[3])**(1/(4-0))-1 13 print(df) 14
#ループ自作(案)
そこでテキストやググって以下のように作ってみましたが、うまくいきませんでした。
なお、ユニークな証券コード(最大の年度)に対して、追加する列を加えていきたいと考えているため、一度groupbyで最新年度だけのデータフレームを作っています。
プログラム自体は動いているのですが、列名の変数展開ができていません。
Python
1 2def make_indicator_multiple(): 3 df = pd.read_csv(Path,encoding="cp932") 4 5 df_only_current = df.groupby('証券コード').max('年度') 6 df['n年前'] = df.groupby('証券コード')['年度'].transform('max')-df['年度'] 7 8 #追加する列名一覧 9 add_columns = ['純資産','売上高','純利益','営業CF','EPS'] 10 11 for column in add_columns: 12 for i in range(0,4): 13 j = i+1 14 df_pvt=df.pivot(index='証券コード',columns='n年前',values='売上高') 15 df_only_current['{column}_{j}→{i}'] = (df_pvt[i]-df_pvt[j])/df_pvt[j] 16 17 print(df_only_current) 18
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2021/10/09 00:56