Python 複数のdfを格納するには（for文、List?）

Question

前提・実現したいこと

dfで取得したデータを４次スプライン補間を行いデータのサンプルアップ、ダウンをしたのち、
再度df型に格納したいのですが

複数のdf（0,1,2）を取得し出力する方法でつまずいています。
最終的には、Excellファイルにdf型で出力したいと考えています。

※Openpyexでの出力は実行できたのですが、出力される形がいまいちだったのでdfで出力をしたいと考えています。よろしくお願いいたします。

該当のソースコード

import pandas as pd
import numpy as np
from scipy.interpolate import interp1d

df = pd.DataFrame({ '0' : [1,2,3,4,5,6,7,8,9,10],
                    '1' : [2,4,6,8,10,12,14,16,18,20],
                    '2' : [1,3,6,9,12,14,16,18,21,24]})

#スプライン補完
def Spline(col):
    Num = df[col].count()
    x = np.arange(Num)
    y = df[col].dropna().values
    f = interp1d(x, y, kind='cubic')
    k = np.linspace(0,Num-1, num=20)
    spline = f(k)
    return spline

#データ出力
#試したこと1
with pd.ExcelWriter(".xlsx") as writer:
    for i in range(0,1):
        pd.DataFrame(Spline(i)).to_excel(writer, sheet_name='Sheet1', index = None)

#試したこと2
def data():
    Spline_list = []
    for i in range(0,2):
        Spline = Spline(i).tolist()
        Spline.append(i)
        i += 1
    return Spline_list

#試したこと2
def data():
    Spline_list = []
    for i in range(0,2):
        Spline = Spline(i).tolist()
        Spline.append(i)
        i += 1
    return Spline_list

試したこと

①エクセルに出力する際に、for文で回してみる
➡︎最後の行(df, 3)のデータのみ出力される

②リストに格納する
➡︎同様に最後の行(df, 3)のデータのみ出力される

Comments

[meg_]

forループの中でシート名を変えてみたらどうでしょうか？

[HK--]

可能ならば同一シートの中に出力したいと考えています。
出来そうでなければそちらの方法を試してみようと思います。
ありがとうございます。

[t_obara]

そもそもSplineで得られる結果は期待通りの内容になっているかは確認されたのですか？

[yureighost]

とりあえず動かしてみたのですが、
Spline関数内でstring型の数字がキーのdfに対して、
数値型の数字でアクセスしようとしているためKeyErrorになってます。
y = df[col].dropna().values

[HK--]

t_obaraさん
掲載する際に変数を変えていたのでミスがありますね。
申し訳ありません、修正を加えます。

Answer 1

スプライン補完の結果がお望みの物かはわかりませんが、
とりあえずこれで辞書リストの'0'、'1'、'2'全ての結果が書き出されます。
辞書型をDataframeでそのまま読ませるとキーがカラム列に並ぶ形になるので、
Excelでもそうなるように変換してますが、
気に入らない場合は、.to_excelの前の.Tを削除してください。

import pandas as pd
import numpy as np
from scipy.interpolate import interp1d

df = pd.DataFrame({ '0' : [1,2,3,4,5,6,7,8,9,10],
                    '1' : [2,4,6,8,10,12,14,16,18,20],
                    '2' : [1,3,6,9,12,14,16,18,21,24]})

#スプライン補完
def Spline(col):
    Num = df[str(col)].count()
    x = np.arange(Num)
    y = df[str(col)].dropna().values
    f = interp1d(x, y, kind='cubic')
    k = np.linspace(0,Num-1, num=20)
    spline = f(k)
    return spline

#データ出力
#試したこと1
with pd.ExcelWriter("output.xlsx") as writer:
    pd.DataFrame([Spline(i) for i in range(0,3)]).T.to_excel(writer, sheet_name='Sheet1', index = None)

#試したこと2
def data():
    return [Spline(i) for i in range(0,3)]

Comments

[HK--]

#試したこと１に修正戴いたものが悩んでいたところでした！
pd.DataFrameとfor文をこのように組み合わせる方法もあるのですね。
勉強させていただきました。ありがとうございました。