【Python】データフレームから特定の列を抽出する(特定の列名は変数で保持する場合)。

Question

やりたいことはデータフレームの特定の列を抽出するという単純なことです。
ただ、一部の列名のリスト?だけを事前に変数として定義して、それを使おうとしています。

import pandas as pd
#original
df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9], 
                   'd': [10, 11, 12], 'e': [13, 14, 15], 'f': [16, 17, 18]})
#dfから列['b', 'e', 'f']を抽出したい
#通常
df1 = df[['b', 'e', 'f']]

#※今回やりたいこと（イメージ）
colname = ['e', 'f']
df2 = df[['b', colname]]

上記のようにdf2を作成すると、エラーメッセージ「unhashable type: 'list'」と出てしまいうまくいきません。
これはcolnameの定義方法を変える必要があるような気がするのですが、正しい書き方が分かりません。
ご教示いただけますと幸いです。

Answer 1

単純に以下でよいですね。['b']というリストとcolnameというリストを+で結合して使います。

df2 = df[['b'] + colname]

Comments

[mashita_takuo]

ご回答ありがとうございます。なるほど、単純に書けるんですね。

Answer 2

import pandas as pd
#original
df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6], 'c': [7, 8, 9], 
                   'd': [10, 11, 12], 'e': [13, 14, 15], 'f': [16, 17, 18]})
#dfから列['b', 'e', 'f']を抽出したい
#通常
df1 = df[['b', 'e', 'f']]

#※今回やりたいこと（イメージ）
colname = ['e', 'f']
colname += 'b'

df2 = df[colname]
df2 = df2.reindex(columns=sorted(df2.columns))
#	b	e	f
# 0	4	13	16
# 1	5	14	17
# 2	6	15	18

Comments

[mashita_takuo]

ご回答ありがとうございました。助かりました。