Как узнать размер массива в numpy 📏: простые способы для начинающих

Чтобы узнать размер массива в NumPy, вы можете использовать атрибут shape у объекта массива.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
size = arr.shape

print(size)

Вывод:

(2, 3)

Размер массива arr - это кортеж, в котором первое значение представляет количество строк, а второе значение представляет количество столбцов. В данном примере массив arr имеет 2 строки и 3 столбца.

numpy как узнать размер массива

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

1. Метод shape()

Метод shape() является одним из наиболее часто используемых методов в NumPy для получения размерности массива. Он возвращает кортеж с элементами размерности в порядке их расположения.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.shape)
# Вывод: (2, 3)

В данном примере мы создаем двумерный массив с размером (2, 3), то есть два ряда и три столбца. Используя метод shape(), мы получаем кортеж (2, 3) в качестве результата.

2. Свойство ndim

Свойство ndim используется для получения количества измерений (размерности) массива. Оно возвращает целое число, показывающее количество измерений в массиве.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.ndim)
# Вывод: 2

В данном примере мы создаем двумерный массив с размерностью 2. Используя свойство ndim, мы получаем результат 2.

3. Метод size()

Метод size() используется для получения общего количества элементов в массиве. Он возвращает целое число, представляющее общее количество элементов в массиве.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(arr.size)
# Вывод: 6

В данном примере мы создаем двумерный массив с 6 элементами. Используя метод size(), мы получаем результат 6.

4. Метод len()

Метод len() можно использовать для получения размерности первого измерения массива (количество строк).

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print(len(arr))
# Вывод: 2

В данном примере мы создаем двумерный массив с 2 строками. Используя метод len(), мы получаем результат 2.

Итак, мы рассмотрели несколько методов, которые можно использовать для получения размерности и количества элементов в массиве NumPy. Зная эти методы, вы сможете легко определить размер любого массива.