DataFrame型のデータをdict型でリストへappendしたい。

Question

test_list = []
name = ['a','b','c']
for n in name:
    df_test = pd.DataFrame([[0,0]],columns=['A','B'])
    dict_df_test = {n:df_test}
    test_list.append(dict_df_test )

DataFrame型の変数をdict型でリストに入れたときにキーでDataFrame型の変数を取り出したいのですが、取り出すとリスト型となってしまいます。DataFrame型のままで取り出すことはできないのでしょうか？

解決策等ありましたら、ご教示頂きたいです。

よろしくお願い致します。

Comments

[jeanbiego]

ルー語みたいになってて分かりづらいのですが、具体的にどの名前の変数をどの型でほしいのか書かれたほうがいいと思います。

[NoPython_Nolife]

大変失礼いたしました。ご指摘ありがとうございます。修正して記載しました。

Answer 1

とりあえず、dict型の値を受け取っている変数dict_df_testが使われていないので、
それをリストに格納すれば一応リストにdict型が入ったような結果にはなります。
ただ出力の結果ですが、こんな形式でいいんですか？

test_list = []
name = ['a','b','c']
for n in name:
    df_test = pd.DataFrame([[0,0]],columns=['A','B'])
    dict_df_test = {n:df_test}
    test_list.append(dict_df_test)

print(test_list)

'''
[{'a':    A  B
0  0  0}, {'b':    A  B
0  0  0}, {'c':    A  B
0  0  0}]
'''

追記

キーでDataFrame型の変数を取り出したい

つまりこういうことですかね？
dict型を更にリストに入れるのではなく、dict型にキーと共に追加していく。
これでキーを指定してDataFrame型を取り出すことができます。

dict_df_test = {}
name = ['a','b','c']
for n in name:
    df_test = pd.DataFrame([[0,0]],columns=['A','B'])
    dict_df_test[n] = df_test

df = dict_df_test['a']
print(df)
'''
   A  B
0  0  0
'''
print(df.dtypes)
'''
A    int64
B    int64
dtype: object
'''

Comments

[NoPython_Nolife]

コメントありがとうございます。コードをいじっている段階でdict_df_test とするところを間違えておりました。おっしゃる通りリストには入るのですが、結果がデータフレーム型ではなくリストでバリューが返ってきてしまいます。

[NoPython_Nolife]

これをもとのデータフレームのままで返してほしいのですが。

[NoPython_Nolife]

御礼が遅れてしまい申し訳ありません。こういったやり方もあるのですね。大変参考になります。ありがとうございました。

Answer 2

test_list = []
name = ['a','b','c']
for n in name:
    df_test = pd.DataFrame([[0,0]],columns=['A','B'])
    dict_df_test = {n:df_test}
    test_list.append(dict_df_test )

質問のコードにてDataFrameのまま追加されていますよ。
何をもって

DataFrame型の変数をdict型でリストに入れたときにキーでDataFrame型の変数を取り出したいのですが、取り出すとリスト型となってしまいます。DataFrame型のままで取り出すことはできないのでしょうか？

と言っているのか分かりませんが、下記の通り取り出せます。

print(test_list[0]['a'])
#   A  B
#0  0  0

print(type(test_list[0]['a']))
#<class 'pandas.core.frame.DataFrame'>

Comments

[NoPython_Nolife]

御礼が遅れてしまい、申し訳ありません。またご指定ありがとうございます。おっしゃる通りでした。大変助かりました。