NumPy zeros()函数的用法

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

zeros() 函数的语法格式如下：

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

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

例如：

import numpy as np
arr1 = np.zeros((3, 3))
print(arr1)

输出结果如下：

[[0. 0. 0.]
[0. 0. 0.]
[0. 0. 0.]]

实际上，np.zeros((3, 3)) 是一种简写方式，它等价于 np.zeros(shape=(3,3))，shape 是数组的形状。

对于 shape 参数来说，它的值是一个元组或列表，所以下面两种方式是等价的。这两种方式在实际开发中都会出现，我们需要了解一下：

# 方式1
arr1 = np.zeros((3, 3))
# 方式2
arr1 = np.zeros([3, 3])

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

# 方式1
arr1 = np.zeros((3, 3))
# 方式2
arr1 = np.zeros((3, 3), dtype=float)

例如：

import numpy as np
arr1 = np.zeros((3, 3), dtype=int)
print(arr1)

输出结果如下：

[[0 0 0]
[0 0 0]
[0 0 0]]

如果想要将数组元素定义成 int 类型，可以使用dtype=int。