Получить список определенных значений столбца Pandas DataFrame

Опубликовано: 27 Марта, 2022

В этой статье мы увидим, как получить все значения столбца в фрейме данных pandas в виде списка. Это может быть очень полезно во многих ситуациях. Предположим, нам нужно получить оценки всех студентов по определенному предмету, получить номера телефонов всех сотрудников и т. Д. Давайте посмотрим, как мы можем этого добиться, на некоторых примерах.

Пример 1. Мы можем иметь все значения столбца в списке, используя метод tolist () .

Syntax: Series.tolist().

Return type: Converted series into List.

Код :

Python3

# import pandas libraey
import pandas as pd
  
# dictionary
dict = {"Name": ["Martha", "Tim",
                "Rob", "Georgia"],
         "Marks": [87, 91
                  97, 95]}
  
# create a dataframe object
df = pd.DataFrame(dict)
  
# show the dataframe
print(df)
  
# list of values of "Marks" column
marks_list = df["Marks"].tolist()
  
# show the list
print(marks_list)

Выход:

Пример 2: Мы увидим, как мы можем получить значения всех столбцов в отдельных списках.

Code:

Python3

# import pandas library
import pandas as pd
  
# dictionary
dict = {"Name": ["Martha", "Tim"
                "Rob", "Georgia"],
        "Marks": [87, 91
                 97, 95]}
  
# create a dataframe object
df = pd.DataFrame(dict)
  
# show the dataframe
print(df)
  
# iterating over and calling 
# tolist() method for 
# each column
for i in list(df):
    
    # show the list of values  
    print(df[i].tolist())

Выход:

Внимание компьютерщик! Укрепите свои основы с помощью базового курса программирования Python и изучите основы.

Для начала подготовьтесь к собеседованию. Расширьте свои концепции структур данных с помощью курса Python DS. А чтобы начать свое путешествие по машинному обучению, присоединяйтесь к курсу Машинное обучение - базовый уровень.