Google colab フォーム入力文字をうまく引数にできない

Question

Google colabのFormを利用し、以下のような入力を設け、
code = ‘’ @param { type:"raw"}

この入力をキーでデータフレームを処理したいと思います。
以下のように ID1 というカラムを入力すると、codeに格納されますので、
code = ‘ID1’ @param { type:"raw"}

code を呼び出して、例えば、以下のようにしています。
df = df.drop(columns=code, axis=1, inplace=True)

できれば、以下のように複数カラムを入力して処理したいのですが、
code = ‘ID’, ‘ID2’ @param { type:"raw"}

ひとつのワードならいけるのですが、2つ以上とすると処理できません。
クオーテーションの処理なのかなんなのか、どなたかご教示いただけますと幸いです。

Answer 1

自己解決しました。
以下のようにフォーム記入した時、
フォーム記入₌ 'a', ’b'

列削除なら、
df.drop(columns=list(フォーム記入),axis=1,inplace=True)

ラベルエンコードなら
for i in フォーム記入:
le = LabelEncoder()
df[i] = le.fit_transform(df[i].astype(str))

渡し方を、適宜変えないといけなかっただけでした。
失礼いたしました。