首页 > 编程笔记 > Python笔记

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

相关文章