Pandas Column to List – Конвертируйте колонку Pandas в список

В этой статье вы узнаете, как преобразовать столбец Pandas в список, что также позволит вам преобразовать серию Pandas в список.

Вы научитесь использовать метод .tolist() библиотеки Pandas, встроенную функцию list() языка Python и метод .tolist() библиотеки NumPy.

Оглавление

• Загрузка образца фрейма данных

• Pandas Column to List – Метод Pandas tolist

• Pandas Series to List – функция Python list()

• Pandas Column to List с использованием Numpy

• Заключение

Загрузка образца фрейма данных

Давайте начнем с загрузки образца dataframe. Скопируйте код ниже, если вы хотите следовать этому руководству.

Код

import pandas as pd
df = pd.DataFrame.from_dict(
    {
        'Name': ['Jim', 'Carl', 'Amy', 'Anne'],
        'Age': [23, 45, 35, 19],
        'Birth City': ['Montreal', 'Paris', 'Toronto', 'Chicago'],
        'Gender': ['M', 'M', 'F', 'F']
    }
)
print(df)

Результат

   Name  Age Birth City Gender
0   Jim   23   Montreal      M
1  Carl   45      Paris      M
2   Amy   35    Toronto      F
3  Anne   19    Chicago      F

Теперь давайте приступим к преобразованию столбца нашего dataframe в список!

Узнайте, как преобразовать список в строку в Python, ознакомившись с моим руководством по этой ссылке: Как преобразовать список в строку в Python.

Pandas Column to List – Метод Pandas tolist

Сначала мы воспользуемся методом Pandas .tolist(), чтобы преобразовать нашу серию Pandas в список. Преимущество этого способа заключается в том, что нам не нужно импортировать библиотеку NumPy, если мы её больше нигде не используем.

Метод Pandas .tolist() применяется к серии Pandas (столбцу) и преобразует её в список.

В качестве примера возьмём столбец name и преобразуем его в список:

>>> names = df['Name'].tolist()
>>> print(names)

['Jim', 'Carl', 'Amy', 'Anne']

Мы можем видеть, что был возвращён список names, содержащий различные значения из этого столбца..

Pandas Series to List – функция Python list()

Аналогично приведённому выше примеру, мы можем использовать встроенную функцию list() языка Python для преобразования серии Pandas в список.

Вместо того чтобы вызывать функцию как метод серии Pandas, мы просто обернём столбец в функцию list().

В следующем примере преобразуем столбец Birth City в список:

>>> cities = list(df['Birth City'])
>>> print(cities)

['Montreal', 'Paris', 'Toronto', 'Chicago']

Аналогично использованию метода Pandas .tolist(), встроенная функция list() возвращает список из столбца Pandas.

Pandas Column to List с использованием Numpy

В качестве последнего метода, рассматриваемого в этом руководстве, вы узнаете, как использовать метод .tolist() библиотеки NumPy.

Сначала мы преобразуем серию Pandas в массив NumPy, используя атрибут .values. Затем вызовем метод .tolist(), чтобы получить список.

В этом последнем примере преобразуем столбец Age в список:

>>> ages = df['Age'].values.tolist()
>>> print(ages)

[23, 45, 35, 19]

Заключение

В этой статье вы узнали, как преобразовать столбец Pandas в список, что также позволяет преобразовать серию Pandas в список. Вы научились делать это с помощью метода tolist() библиотеки Pandas, встроенной функции list() языка Python и метода tolist() библиотеки NumPy. Чтобы узнать больше о методе tolist() в Pandas, ознакомьтесь с официальной документацией по этой ссылке. Чтобы узнать больше о встроенной функции list() в Python, посетите официальную документацию по этой ссылке.