NumPy 创建数组


NumPy 创建数组

NumPy 是 Python 语言的一个扩展程序库,支持大量高维数组与矩阵运算,是 Python 科学计算的重要工具之一。在 NumPy 中,最基本的对象是数组 ndarray,数组中的元素必须是相同的类型。在本文中,我们将介绍 NumPy 创建数组的方法和细节。

1. 创建空数组

创建一个空的数组,只需要指定数组的形状即可。形状可以是一个整数元组,指定每一个维度的大小。

import numpy as np

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

输出:

[[0. 1. 2.]
 [3. 4. 5.]]

2. 创建全0数组

创建一个全0数组,只需要调用 np.zeros() 函数,并传入形状即可。和前面一样,形状是一个整数元组。

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

输出:

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

3. 创建全1数组

创建一个全1数组,只需要调用 np.ones() 函数,并传入形状即可。和前面一样,形状是一个整数元组。

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

输出:

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

4. 创建连续数组

通过调用 np.arange() 函数可以创建一个连续的数组。np.arange(start, stop, step) 函数的三个参数分别表示开始值、结束值、和步长。不同于 Python 自带的 range() 函数,np.arange() 能够处理浮点数步长。

d = np.arange(0, 1, 0.2)  # 创建一个起始于 0,以 0.2 为步长,小于 1 的连续数组
print(d)

输出:

[0.  0.2 0.4 0.6 0.8]

5. 创建随机数组

通过调用 np.random.rand() 函数可以创建一个随机的数组。np.random.rand(d0, d1, ..., dn) 函数可以根据形状创建一个指定范围的随机数组。

e = np.random.rand(2, 3)  # 创建一个 2 行 3 列的随机数组
print(e)

输出:

[[0.30266629 0.08886028 0.55484194]
 [0.27401813 0.52388355 0.48962655]]

以上就是 NumPy 创建数组的几种方法。除此之外,NumPy 还提供了许多方式来创建、填充数组。不同的创建方式适用于不同的场景,可以根据需要灵活选择。