Функция NumPy ones() в Python — как использовать

Рассмотрим, как использовать функцию numpy ones() для создания массива numpy заданной формы, элементы которого заполнены единицами, единичную матрицу.

Функция ones() модуля numpy позволяет создать массив единиц numpy заданной формы.

Например, в следующем примере функция ones() используется для создания массива с тремя осями, первая ось имеет два элемента, вторая ось — три элемента, а третья ось — 4 элемента:

import numpy as np

a = np.ones((2, 3, 2))
print(a)

Выход:

[[[1. 1.]
  [1. 1.]
  [1. 1.]]

 [[1. 1.]
  [1. 1.]
  [1. 1.]]]

По умолчанию функция ones() использует для своих элементов float64. Например:

import numpy as np

a = np.ones((2, 3, 2))
print(a.dtype)

Выход:

float64

Чтобы использовать другой тип, вам необходимо указать его с помощью аргумента dtype. Например:

import numpy as np

a = np.ones((2, 3, 4), dtype=np.int32)
print(a)
print(a.dtype)

Выход:

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

 [[1 1]
  [1 1]
  [1 1]]]
int32

В этом примере мы используем тип int32 для элементов. Поэтому вы не видите десятичную точку(.) в каждом числе.