Вывести колонки фрейма из массива индексов
Как вывести только те колонки фрейма, индексы которых содержатся в массиве или списке? Например:
Фрейм:
print(df)
a b c
0 1 4 7
1 2 5 8
2 3 6 9
Массив:
arr = np.array([0,1])
Ожидаемый результат:
print(df)
a b
0 1 4
1 2 5
2 3 6