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

Количество строк и столбцов в Python можно посчитать с помощью функций len() и len() соответственно.

# Пример
матрица = [[1, 2, 3],
           [4, 5, 6],
           [7, 8, 9]]

количество_строк = len(матрица)
количество_столбцов = len(матрица[0])  # Предполагаем, что матрица не пустая

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

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

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

1. Подсчет строк и столбцов в списках и кортежах

Если у вас есть список или кортеж, вы можете использовать функции len() и len() соответственно, чтобы подсчитать количество элементов в них. Для подсчета строк в списке или кортеже используйте функцию len(), а для подсчета столбцов - число элементов в каждом из них.

    my_list = ['строка 1', 'строка 2', 'строка 3']
    num_rows = len(my_list)
    num_columns = len(my_list[0])
    

В данном примере мы имеем список my_list, состоящий из трех строк. Переменная num_rows будет содержать значение 3, так как у нас три строки, а переменная num_columns будет содержать значение длины первой строки, то есть число столбцов.

2. Подсчет строк и столбцов в массивах NumPy

Если у вас есть массив NumPy, вы можете использовать атрибуты shape и ndim для подсчета строк и столбцов. Атрибут shape возвращает кортеж, содержащий размерность массива, где первый элемент - количество строк, а второй - количество столбцов. Атрибут ndim возвращает количество осей массива.

    import numpy as np
    
    my_array = np.array([[1, 2, 3], [4, 5, 6]])
    num_rows = my_array.shape[0]
    num_columns = my_array.shape[1]
    num_dimensions = my_array.ndim
    

В данном примере мы объявляем массив my_array размером 2x3. Переменные num_rows, num_columns и num_dimensions будут содержать соответственно значения 2, 3 и 2.

3. Подсчет строк и столбцов в таблицах Pandas

Если у вас есть таблица Pandas, вы можете использовать атрибуты shape и columns для подсчета строк и столбцов. Атрибут shape возвращает кортеж, содержащий количество строк и столбцов таблицы. Атрибут columns возвращает список с именами столбцов таблицы.

    import pandas as pd
    
    data = {'Страна': ['Россия', 'США', 'Китай'],
            'Население': [144.5, 328.2, 1393.8]}
    df = pd.DataFrame(data)
    num_rows = df.shape[0]
    num_columns = df.shape[1]
    column_names = df.columns.to_list()
    

В данном примере мы создаем таблицу Pandas с двумя столбцами "Страна" и "Население". Переменные num_rows и num_columns будут содержать соответственно значения 3 и 2, так как у нас три строки и два столбца. Переменная column_names будет содержать список со значениями ["Страна", "Население"].

Вывод

Мы рассмотрели несколько способов подсчета количества строк и столбцов в Python для различных структур данных, таких как списки, кортежи, массивы NumPy и таблицы Pandas. Вы можете использовать соответствующие функции и атрибуты в зависимости от типа данных, с которыми вы работаете.

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