Python numpy.array()生成相同元素数组的示例

生成相同元素的numpy数组可以使用numpy.array()函数。我们来看一下生成相同元素的numpy数组的两个示例。

示例1：生成全0 numpy数组

我们要生成一个5行3列的全0数组。看下面的代码：

import numpy as np

a = np.zeros((5,3))
print(a)

输出结果：

array([[0., 0., 0.],
       [0., 0., 0.],
       [0., 0., 0.],
       [0., 0., 0.],
       [0., 0., 0.]])

这里我们使用了numpy的zeros函数生成一个具有给定形状和数据类型的全0数组。我们需要传递一个元组 (5, 3)，指定要生成的数组的形状。输出结果为一个元素全部为0的5行3列的数组。

示例2：生成全1 numpy数组

我们要生成一个3行4列的全1数组。看下面的代码：

import numpy as np

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

输出结果：

array([[1., 1., 1., 1.],
       [1., 1., 1., 1.],
       [1., 1., 1., 1.]])

这里我们使用numpy的ones函数生成一个具有给定形状和数据类型的全1数组。传递一个元组 (3, 4)，指定要生成的数组的形状。输出结果为一个元素全部为1的3行4列的数组。