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

Как задать размерность массива numpy?

Чтобы задать размерность массива в библиотеке numpy, вы можете использовать функцию numpy.reshape(). Эта функция позволяет изменить форму массива без изменения его данных.

Вот пример:

import numpy as np

# Исходный массив
arr = np.array([1, 2, 3, 4, 5, 6])

# Задаем размерность (2 строки, 3 столбца)
reshaped_arr = np.reshape(arr, (2, 3))

print(reshaped_arr)

Этот код создает исходный массив arr с 6 элементами и затем изменяет его размерность до 2 строк и 3 столбцов с помощью np.reshape(). Результат выводится на экран.

Как задать размерность массива numpy

Когда мы работаем с массивами в библиотеке numpy, очень важно правильно задать размерность этого массива. В этой статье мы рассмотрим различные способы задания размерности массива numpy и предоставим примеры кода для каждого способа.

1. Задание размерности с использованием параметра shape

Один из способов задания размерности массива numpy - это использование параметра shape. Параметр shape принимает кортеж, в котором перечисляются размеры массива по каждой оси.

import numpy as np

# Задаем размерность массива
arr = np.array([[1, 2, 3], [4, 5, 6]], dtype=np.int32)
arr.shape = (3, 2)

print(arr)
# Вывод: 
# [[1 2]
#  [3 4]
#  [5 6]]

В приведенном примере мы создаем двумерный массив с помощью функции np.array, а затем изменяем его размерность с помощью arr.shape = (3, 2).

2. Задание размерности с использованием функции reshape

Другой способ задания размерности массива numpy - это использование функции reshape. Функция reshape возвращает новый массив с указанной размерностью.

import numpy as np

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

# Используем функцию reshape для задания размерности
new_arr = arr.reshape((2, 3))

print(new_arr)
# Вывод:
# [[1 2 3]
#  [4 5 6]]

В данном примере мы создаем одномерный массив, а затем с помощью функции reshape меняем его размерность на двумерный массив размером 2x3.

3. Создание массива с заданной размерностью

Третий способ задания размерности массива numpy - это создание массива с заданной размерностью с использованием функции np.zeros или np.ones. Функция np.zeros создает массив, заполненный нулями, заданной формы, а функция np.ones создает массив, заполненный единицами, заданной формы. Мы можем задать размерность массива, указав кортеж в качестве параметра shape.

import numpy as np

# Создаем массив нулей с размерностью 3x3
arr_zeros = np.zeros((3, 3))

print(arr_zeros)
# Вывод:
# [[0. 0. 0.]
#  [0. 0. 0.]
#  [0. 0. 0.]]

# Создаем массив единиц с размерностью 2x4
arr_ones = np.ones((2, 4))

print(arr_ones)
# Вывод:
# [[1. 1. 1. 1.]
#  [1. 1. 1. 1.]]

В данном примере мы создаем массив нулей размером 3x3 с помощью функции np.zeros и массив единиц размером 2x4 с помощью функции np.ones.

Теперь вы знаете, как задать размерность массива numpy с помощью параметра shape, функции reshape и функций np.zeros и np.ones. Надеюсь, эта статья была полезна для вас!