Как правильно вычислить количество столбцов в pandas? 📊

Количество столбцов в объекте DataFrame в библиотеке Pandas можно посчитать с помощью следующего кода:

import pandas as pd

data = {'Column1': [1, 2, 3],
        'Column2': ['a', 'b', 'c'],
        'Column3': [True, False, True]}

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

print(f"Количество столбцов: {column_count}")

В этом примере мы импортируем библиотеку Pandas и создаем объект DataFrame с данными в словаре data. Затем мы используем метод columns для получения списка столбцов и функцию len() для подсчета их количества. Результат выводится на экран.

Как посчитать количество столбцов в Pandas

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

Способ 1: Использование атрибута shape

Самый простой способ узнать количество столбцов - это использовать атрибут shape вашего DataFrame. Атрибут shape возвращает кортеж, состоящий из двух значений: количество строк и количество столбцов. Чтобы получить только количество столбцов, вы можете обратиться к второму элементу этого кортежа.

import pandas as pd

# Создание DataFrame для примера
data = {'Имя': ['Анна', 'Иван', 'Мария'],
        'Возраст': [25, 30, 28],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург']}

df = pd.DataFrame(data)

# Использование атрибута shape для подсчета количества столбцов
num_columns = df.shape[1]

print("Количество столбцов:", num_columns)
    

Результат выполнения кода:

Количество столбцов: 3

Способ 2: Использование функции len()

Другой способ узнать количество столбцов - это использовать функцию len() с атрибутом columns. Атрибут columns возвращает список имен столбцов в DataFrame, а функция len() возвращает количество элементов в этом списке.

import pandas as pd

# Создание DataFrame для примера
data = {'Имя': ['Анна', 'Иван', 'Мария'],
        'Возраст': [25, 30, 28],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург']}

df = pd.DataFrame(data)

# Использование функции len() для подсчета количества столбцов
num_columns = len(df.columns)

print("Количество столбцов:", num_columns)
    

Результат выполнения кода:

Количество столбцов: 3

Способ 3: Использование метода info()

Третий способ узнать количество столбцов - это использовать метод info(). Метод info() выводит общую информацию о DataFrame, включая количество столбцов, типы данных и количество ненулевых значений для каждого столбца.

import pandas as pd

# Создание DataFrame для примера
data = {'Имя': ['Анна', 'Иван', 'Мария'],
        'Возраст': [25, 30, 28],
        'Город': ['Москва', 'Санкт-Петербург', 'Екатеринбург']}

df = pd.DataFrame(data)

# Использование метода info() для отображения информации о DataFrame
df.info()
    

Результат выполнения кода:

RangeIndex: 3 entries, 0 to 2
Data columns (total 3 columns):
 #   Column   Non-Null Count  Dtype 
---  ------   --------------  ----- 
 0   Имя      3 non-null      object
 1   Возраст  3 non-null      int64 
 2   Город    3 non-null      object
dtypes: int64(1), object(2)
memory usage: 200.0+ bytes
    

В данном выводе информации можно увидеть строку Data columns (total 3 columns), где указано общее количество столбцов.

Заключение

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