Выбор нескольких столбцов из фрейма данных с использованием словаря

Question

У меня есть фрейм данных с именами столбцов

** RI Na Mg Al Si K Ca Ba Fe Тип **

Тип столбца является категориальным и состоит из 4 типов. Как извлечь только столбцы RI, Al, Si, Ba с типом 2, используя словарь?

Я пробовал использовать следующий код:

pd.DataFrame.from_dict(data=df[df['Type']==2], orient='index',columns=['RI','Al','Si','Ba']) 

Я получаю ошибку:

'numpy.ndarray' object is not callable

Как выбрать только указанные c столбцы типа 2?

Answer 1

Сначала отфильтруйте строки фрейма данных с типом, равным 2, отфильтрованным_df = df [df [Тип == 2]] ... Теперь преобразуйте отфильтрованный df в dict с помощью to_dict ()
dict = Filter_df. to_dict () Теперь из dict вы можете отфильтровать несколько столбцов, передав нужные столбцы в качестве ключей key = ['Ri', 'Ai', 'Si', 'Ba'] для ввода ключей: dict.get (key)