温馨提示×

怎么创建NumPy的零数组和单位数组

小亿
102
2024-05-11 16:28:49
栏目: 编程语言

要创建NumPy的零数组和单位数组,可以使用numpy.zeros()numpy.eye()函数。

  1. 创建一个形状为(3,3)的零数组:
import numpy as np

zero_array = np.zeros((3,3))
print(zero_array)

输出:

[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
  1. 创建一个3x3的单位数组:
import numpy as np

eye_array = np.eye(3)
print(eye_array)

输出:

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

这样就可以创建出指定形状的零数组和单位数组。

0