NumPy ones()的用法

在 NumPy 中，可以使用 ones() 方法来创建一个元素值全为 1 的数组。

ones() 方法的语法格式如下：

np.ones((a, b, ..., n), dtype=int或float)

ones() 可以接收两个参数，第 1 个参数是一个元组，它是一个必选参数，表示创建一个 a×b×…×n 的数组；第 2 个参数是一个可选参数，它用于定义元素的类型，默认是 float 类型。

例如：

import numpy as np
arr = np.ones((3, 3))
print(arr)

输出结果为：

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

默认情况下，使用 ones() 创建的数组的元素是 float 类型，以下两种方式是等价的：

# 方式1
arr2 = np.ones((3, 3))
# 方式2
arr2 = np.ones((3, 3), dtype=float)

如果想要将数组元素定义成 int 类型，可以使用 dtype=int，再举个例子：

import numpy as np
arr = np.ones((3, 3), dtype=int)
print(arr)

输出结果如下：

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