Python中的np.zeros()和np.ones()函数详解

np.zeros()和np.ones()都是NumPy中用于创建数组的函数，本文将详细介绍其用法。

np.zeros()

np.zeros()函数用于创建所有元素都为0的数组。函数的语法如下：

numpy.zeros(shape, dtype=float, order='C')

参数说明：

• shape：数组的形状，可以是一个整数，表示创建一个n维度的数组；也可以是一个元组或列表，每个元素表示各维度的长度。
• dtype：可选，默认为float，表示数组的数据类型。
• order：可选，默认为C，表示多维数组的元素在内存中排列方式，C表示行优先，F表示列优先。

示例1：

import numpy as np

# 创建一个2行3列的全零数组
a = np.zeros((2, 3))
print(a)

输出结果为：

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

示例2：

import numpy as np

# 创建一个5维的全零数组
a = np.zeros(5)
print(a)

输出结果为：

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

np.ones()

np.ones()函数用于创建所有元素都为1的数组。函数的语法如下：

numpy.ones(shape, dtype=None, order='C')

参数说明：

• shape：数组的形状，可以是一个整数，表示创建一个n维度的数组；也可以是一个元组或列表，每个元素表示各维度的长度。
• dtype：可选，默认为None，表示数组的数据类型。
• order：可选，默认为C，表示多维数组的元素在内存中排列方式，C表示行优先，F表示列优先。

示例1：

import numpy as np

# 创建一个2行3列的全1数组
a = np.ones((2, 3))
print(a)

输出结果为：

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

示例2：

import numpy as np

# 创建一个5维的全1数组
a = np.ones(5)
print(a)

输出结果为：

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

以上就是np.zeros()和np.ones()函数的详细介绍，希望能对你有所帮助。