Python Pandas: выбор столбцов по именам столбцов

Question

Я могу выбрать столбцы DataFrame Pandas с их позициями:

df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6], "c": [7, 8, 9],
                   "d": [10, 11, 12], "e": [13, 14, 15]})
df.iloc[:, 1:4]

С этим я могу выбрать столбцы b до d.

Есть ли простой способ сделать это, используя имена столбцов? Что-то вроде:

df.SOME_FUNCTION_OR_A_SPECIFIC_SYNTAX("b", "d")

Answer 1

В этом случае вы также можете использовать regex:

df = df.filter(regex='[b-d]')

Или более общий, указав имя столбца:

df = df[['b','c','d']]

Answer 2

Используйте loc только с : для диапазона столбцов:

df1 = df.loc[:, 'b':'d']
print (df1)

   b  c   d
0  4  7  10
1  5  8  11
2  6  9  12