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

Question

У меня есть датафрейм с заданными именами столбцов, и я хочу вернуть столбец с указанным именем:

name_of_column = 'name1'  # string variable

Я пытался использовать это:

dataframe.iloc[:, name_of_column]

Но это не сработало. Что я должен делать?

Answer 1

Используйте loc вместо iloc, и ваш синтаксис будет работать.iloc для индексации по целочисленной позиции (это то, что обозначает i), а loc для индексации по метка .Таким образом, вы можете использовать:

dataframe.loc[:, name_of_column]

Сказав это, более обычным способом получения серии является непосредственное использование __getitem__:

dataframe[name_of_column]

Answer 2

Вы можете просто сделать:

dataframe[column_name]

Выберет столбец.

iloc() метод находит предмет в пандах по индексу.

Дополнительные примеры данных выбора, которые можно найти в Индексация и выбор данных Pandas