🔎 Как вывести название колонок в pandas? Учимся вместе! 🐼

Чтобы вывести названия колонок в Pandas, вы можете использовать атрибут columns у объекта DataFrame. Вот пример:

import pandas as pd

# Создание DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)

# Вывод названий колонок
print(df.columns)

В этом примере мы создали DataFrame 'df' с двумя колонками 'A' и 'B'. Затем мы использовали атрибут columns для вывода названий колонок. Вы увидите результат:

Index(['A', 'B'], dtype='object')

Как вывести название колонок в pandas

В библиотеке Pandas, которая широко используется для анализа данных на языке Python, есть несколько способов вывести названия колонок в DataFrame. В этой статье рассмотрим несколько методов, которые помогут вам достичь этой цели.

Метод 1: Использование атрибута columns

Первый способ - использовать атрибут columns DataFrame. Этот атрибут возвращает список со всеми названиями колонок. Рассмотрим пример:

import pandas as pd

# Создаем DataFrame
data = {'Name': ['John', 'Peter', 'Sara'],
        'Age': [25, 30, 35],
        'City': ['Moscow', 'New York', 'London']}

df = pd.DataFrame(data)
columns = df.columns

print(columns)
    

Вывод программы:

'Index(['Name', 'Age', 'City'], dtype='object')

Как видно, мы получили список с названиями всех колонок в DataFrame. Примечательно, что тип данных этого списка - Index.

Метод 2: Преобразование атрибута columns в список

Второй способ состоит в преобразовании атрибута columns в обычный список с помощью метода tolist(). Рассмотрим код:

import pandas as pd

# Создаем DataFrame
data = {'Name': ['John', 'Peter', 'Sara'],
        'Age': [25, 30, 35],
        'City': ['Moscow', 'New York', 'London']}

df = pd.DataFrame(data)
columns_list = df.columns.tolist()

print(columns_list)
    

Вывод программы:

['Name', 'Age', 'City']

Теперь мы получили обычный список, который может быть использован в дальнейшем коде.

Метод 3: Использование метода get_option('display.max_columns')

Третий способ - использовать метод get_option('display.max_columns') для получения максимального количества отображаемых колонок в выводе. Затем можно использовать метод head() для отображения первых нескольких строк DataFrame, в которых будут видны названия колонок. Рассмотрим пример:

import pandas as pd

# Создаем DataFrame
data = {'Name': ['John', 'Peter', 'Sara'],
        'Age': [25, 30, 35],
        'City': ['Moscow', 'New York', 'London']}

df = pd.DataFrame(data)
max_columns = pd.get_option('display.max_columns')
df_head = df.head()

print(df_head)
    

Вывод программы:

   Name  Age      City
0  John   25    Moscow
1 Peter   30  New York
2  Sara   35    London
    

Как видно из вывода, рядом с данными отображаются названия колонок. Обратите внимание, что если количество колонок в DataFrame больше, чем значение параметра display.max_columns, названия остальных колонок будут заменены многоточием (...).

В заключении

В этой статье мы рассмотрели три способа вывести названия колонок в библиотеке Pandas. Вы можете выбрать подход, который лучше всего соответствует вашим потребностям. Удачи в работе с Pandas!