python 順次抽出したデータフレームを一つのデーターフレームにまとめる方法を教えてください。

Question

前提・実現したいこと

python

for文を使って順次作ったデーターフレームを１つのデーターフレームにまとめていく方法を教えて頂けないでしょうか？

１列ごとの要素から予測したデータを集約したいのですが、うまく出来ず苦戦しております。

せいぜいprintで個々に表示させることはできたのですが、その後前に進めません。

宜しくお願い致します。

■■な機能を実装中に以下のエラーメッセージが発生しました。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

import pandas as pd
import datetime as dt
#Prophet　インストール
!pip3 install pystan
!pip3 install fbprophet
from fbprophet import Prophet

df_sample = pd.DataFrame({'ds': [42461,42462,42463,42464,42465],
                    'B': [25,26,27,28,29],
                    'C': [110,111,112,113,114],
                    'D': [125,126,127,128,129],
                    'E': [222,224,226,228,230]},
                   index=['ONE', 'TWO', 'THREE','FOUR','FIVE'])

#シリアル値を変換　閏年補正あり
df_sample['ds']=pd.to_timedelta(df_sample['ds'].astype('float') - 2, unit='D') + pd.to_datetime('1900/1/1')

#関数でできないか
i = 1
def func():     
    _df = df_sample.iloc[:,[0,i]]

#prophetの書式に揃える'y''ds'
    _df.columns = ['ds','y'] 

#prophetへ投入
#モデルの構築
    model = Prophet()
    model.fit(_df)

#5日先まで予測
    future = model.make_future_dataframe(5,freq='d')
    forecast_d = model.predict(future)
    forecast_agg = forecast_d.iloc[:,[0,-1]]

#forecastに_dfを結合
    df_merge = pd.merge(forecast_agg,_df,on='ds',how='left')

#columnの並べ替えと必要事項のみ
    df_m = df_merge.iloc[:,[0,1,-1]]

#シンプルに結合できるようにもとのdfを必要列のみに絞る
    df_s = df_sample.iloc[:,[0,i]]

#meage結合
    df_m_merge = pd.merge(df_m,df_s,on='ds',how='left')
    df_drop = df_m_merge.drop('y',axis=1)
    print(df_drop)

#printで表示
for i in range(4):
    i += 1
    print(func())

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

全てのデータフレームのカラムが同じならば、以下のようにしてみてください。
同じではない場合は事前に揃える必要があります。

まず、funcを以下のように変更してください。
def func(i): #引数でiを渡す。
print(df_drop) を
return df_drop に変更する。

total_df = pd.concat([func(i) for i in range(1, 5, 1)])
で結合したデータフレームがtotal_dfに入るはずです。

Comments

[jimmyjimmy]

初めての投稿だったのですが、即答に感激しております。

concatのなかに[func(i) for i in range(1, 5, 1)]にいれる発想とかどのように考えればよいのでしょうか？

ネット検索してもpd.concat([s1, s2], axis=1)ぐらいで、せいぜい列方向に程度しか検索できずにいました。数日進めず今回投稿にチャレンジしてみました。

せっかくご回答頂けましたので色々いじって試してみたいと思います。

本当にありがとうございました。

[ppaul]

concatのなかに[func(i) for i in range(1, 5, 1)]にいれる発想

こういうのは慣れといえば慣れですね。
アイスクリームを食べたいけれども冷蔵庫がない、だから冷蔵庫を買おうという方法だけでなく、コンビニの隣に引っ越せば冷蔵庫はいらないとか、雪印に就職するのはどうかとか、実現性はともかくとして、複数の答えを考えるのを頭の体操としてやってみるのがいいかもしれません。

ただ、手持ちの情報から出来ることとやりたいことの間が直接につながらないときは、いろいろな可能性、いろいろな組み合わせを考えるという習慣を身につけておくと、プログラミング以外でも役に立ちます。