Функция ones_like

Функция ones_like создает новый массив NumPy, заполненный единицами, форма и тип данных которого совпадают с переданным массивом. Первым параметром функция принимает исходный массив. Вторым параметром можно указать тип данных для нового массива.

Синтаксис

np.ones_like(prototype, [dtype])

Пример

Давайте создадим массив из единиц, форма которого совпадает с формой одномерного массива из пяти элементов:

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
res = np.ones_like(arr)
print(res)

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

[1 1 1 1 1]

Пример

Создадим массив из единиц на основе двумерного массива:

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
res = np.ones_like(arr)
print(res)

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

[[1 1 1]
 [1 1 1]]

Пример

Если нужно изменить тип данных элементов результирующего массива, укажите параметр dtype:

import numpy as np

arr = np.array([1, 2, 3])
res = np.ones_like(arr, dtype=np.float32)
print(res)
print(res.dtype)

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

[1. 1. 1.]
float32

Пример

Массив ones_like работает не только с числами, но и с другими типами данных, например, со строками:

import numpy as np

arr = np.array(['a', 'b', 'c', 'd', 'e'])
res = np.ones_like(arr, dtype='U1')
print(res)

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

['1' '1' '1' '1' '1']

Смотрите также

функцию zeros_like,
которая создает массив из нулей по образцу
функцию ones,
которая создает массив из единиц заданной формы
функцию empty_like,
которая создает пустой массив по образцу
функцию full_like,
которая создает массив из любого значения по образцу