python リスト名を変数で指定する

Question

実現したいこと

df.queryで抽出してきたリスト型のデータを既存のリストへ結合する。これをfor分で回したい。

発生している問題・分からないこと

queryを使ってほしいデータを抽出してくることはできましたが、それをリストに格納することができません。
４行目でリストを指定する際、リスト型でなく文字列型と認識されてるためと理解できたのですが、回避策が分かりませんでした。

該当のソースコード

list_0=[],list_1=[],list_2=[]
for i in range(3):
    list_tmp = df.query("@i<A<@i+1")["B"].to_list()
    "list_" + str(i).extend(list_tmp)

試したこと・調べたこと

• teratailやGoogle等で検索した
• ソースコードを自分なりに変更した
• 知人に聞いた
• その他

上記の詳細・結果

「変数を用いたリストの指定」について有用な情報にたどり着くことができませんでした。

補足

特になし

Answer 1

正解付いていますが、補足です。

「list_0=[],list_1=[],list_2=[]」のような、番号付きの同じようなデータを複数使いたくなったら、そのときは、それをリストで表現すべき場面です。 リスト中に入れるのはリストであっても変わりません。

list_of_list = [[], [], []]

こんな感じです。

また、単に空のリストを用意して、そこに追加するのでもいいでしょう。

 list_of_list = []
for i in range(3):
    list_tmp = df.query("@i<A<@i+1")["B"].to_list()
    list_of_list.append(list_tmp)

Comments

[hiroki]

申し訳ありません、ベストアンサーつけ間違えてしまいました。

構文？関数？についての知識もそうですが、できる人の思考過程を少しでも身につけたいところです。
今回はリストを格納したリストです。参考になりました、ありがとうございました。

Answer 2

抽出してくることはできましたが、それをリストに格納することができません。

以下はリストのリストを使う場合です。(list という名前はクラス名として使われているため lst に変更しています)

import pandas as pd

df = pd.DataFrame({
    'A': [2.5, 0.5, 1.5],
    'B': [4, 5, 6],
})

N = 3
lst = [[] for _ in range(N)]
for i in range(N):
    lst_tmp = df.query("@i<A<@i+1")["B"].to_list()
    lst[i].extend(lst_tmp)

print(lst)

# [[5], [6], [4]]

Comments

[hiroki]

DataFrameだったり辞書だったりリストだったりいろいろ試したつもりでしたが、
リストに各リストを格納する方法は思いつきませんでした。
解決しました、ありがとうございます。

[hiroki]

申し訳ありません、ベストアンサーつけ間違えてしまいました。

[melian]

一応コメントしておきますと、BAの変更は可能です。