execを関数に組み込みたい

Question

前提

csvファイルをデータフレームに読み込む際に、ファイル名ごとに列選択して読み込みたい

これを実現するためにexec関数を用いたところ、エラーは出力されませんがexec関数が通っていないという事象が起こりました。

実現したいこと

csvファイルをデータフレームに読み込む際に、ファイル名ごとに列選択して読み込みたい

該当のソースコード1(正しく動くコード)

test_arr = [column1,column2]

test= "テスト"
df = pd.read_csv(FOLDER+test+".csv", sep=',')
print("選択前:" + str(len(df.columns)))　#出力 100
exec_command = "df = df["+test+"_arr]"
exec(exec_command)
print("選択後:" + str(len(df.columns)))　#出力 2

#以下dfを操作し可視化を行いますが、本題と関係がないので省略します

#出力
選択前:100
選択後:2

該当のソースコード2(正しく動かないコード)

test_arr = [column1,column2]

def select(test):
     df = pd.read_csv(FOLDER+test+".csv", sep=',')
     print("選択前:" + str(len(df.columns)))
     exec_command = "df = df["+test+"_arr]"
     exec(exec_command)
     print("選択後:" + str(len(df.columns)))
     #以下省略

test = "テスト"
select(test)

#出力
選択前:100
選択後:100

試したこと

関数として定義するとexecが動いていないことが確認できていますが
exec関数を使うのが初めてなこともあり、原因がわかりません

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

Windows11
VSCodeでipynbファイルで動かしています

Comments

[melian]

read_csv() には usecols というキーワードがあって、読み込むカラムを指定することができます。

df = pd.read_csv(FOLDER+test+".csv", sep=',', usecols=test_arr)

[754x]

ありがとうございます。

Answer 1

求めているものとは違う方向の回答になってしまいますが、こうすべきと思うので。

質問からすると、

ほげ_arr = [column1,column2]
ふが_arr = [column1,column2]

for txt in ["ほげ", "ふが"]
   df = df[txt_arr]

のように使いたいけれど、変数に入った文字列(ほげとかふが)から対応する変数名を作りたいということで、execを持ち出したということだと思います。
ですが、この考えは筋が悪い悪手です。

連番の入った変数や、変数を使って変数を切り替えたい場面に出会った場合、元のデータにリスト(番号の場合)や辞書(文字列の場合)を使うのが正しい筋です。

list_info = {}
list_info['ほげ'] = [column1,column2]
list_info['ふが'] = [column1,column2]

for txt in ["ほげ", "ふが"]
   df = df[list_info[txt]]

のようにしたほうがいいでしょう。

Comments

[754x]

早速の回答ありがとうございます。
実現したいことを達成できたため、ベストアンサーにさせていただきました。
ありがとうございました。