Numpy ndarray 对象

Numpy ndarray 对象

本小节将详述如何创建 ndarray 对象,以及如何创建特殊的 ndarray 对象。

1. ndarray 对象

ndarray,也就是我们常用的 N 维数组对象,是 Numpy 最核心的组成部分。你可以把它视为一个快速而灵活的大数据集容器,利用这种数组你可以便捷地执行一些数学运算。

ndarray 有两个显著特点:

  • ndarray 是一个通用的同构数据多维容器,也就是说,该容器中的每一个元素都必须是相同类型的,且每个元素在内存中都有相同存储大小的区域;

  • ndarray 具有矢量算术运算能力和复杂的广播能力,并具有执行速度快和节省空间的特点。

2. 创建 ndarray

2.1 numpy.array 函数

创建数组对象最简单的方法是利用 array 函数,该函数的输入为序列型对象,输出为一个含有传入数据的 Numpy 数组。 array 函数的常用调用参数如下:

numpy.array(object, dtype=None, copy=True, ndmin=)
参数说明
objectarray的主要输入参数,可以是数组、有序序列,或者是嵌套的序列
dtype数据类型,用来指定生成的ndarray数据结构的元素类型
copy对象是否被复制,默认为True
ndmin指定返回数组的最小维数

Tips:关于数据类型相关知识,我们会在下个小节详细讲解。

案例

以列表转换为例:

import numpy as np

data0 = [, , , , ]arr0 = np.array(data0)arr0

这里通过 array 函数,输入列表,输出一个数组对象。

Out:array([, , , , ])

在上述案例中,并未指定dtype。这是因为在通常情况下,np.array函数会尝试为新建的数组推断出一个合理的数据类型:

arr0.dtype
Out:dtype('int32')

可以看到自动创建的数据类型为int32

案例

当然了,你可以在开始的时候,利用dtype参数,显式地指定数组元素的数据类型。这里我们以一个嵌套列表为例,进行说明:

data1 = [[, , ], [, , ], [, , ]]arr1 = np.array(data1, dtype='float')arr1

这里通过 array 函数,输入嵌套列表,并同时指定数据类型为 float,输出对应的数组对象。

Out:array([[. , . , ],   [. , , . ],   [, . , . ]])

查看数组对象的数据类型为float64。

arr1.dtype
Out:dtype('float64')

案例

也可以利用 ndmin 参数,手动指定创建数组的维数:

arr2 = np.array(data0, ndmin=)arr2
Out:array([[, , , , ]])

这里利用 ndmin 参数,手动创建了一个二维数组。

3. 常用的 ndarray 创建函数

np.array 之外,Numpy 内置了一些其他方法,可以快速创建一些具有特殊特征的数组。比如,利用 zeros 函数和 ones 函数可以创建一个指定大小的全 0 或全 1 的数组。

3.1 numpy.zeros 函数

zeros 函数可以创建一个指定大小的、以 0 来填充的数组。

案例

创建一个长度为 4 的全 0 数组:

np.zeros()
Out:array([., ., ., .])

3.2 numpy.ones 函数

ones 函数可以创建一个指定大小的、以 1 来填充的数组。

案例

创建一个大小为 3×3 的全 0 数组方阵:

np.ones((,))
Out:array([[., ., .],   [., ., .],   [., ., .]])

需要注意的是,对于创建 2 维及以上的数组时,zeros函数和 ones 函数接收一个表示数组大小的元组 tuple 作为输入。

3.3 numpy.empty 函数

类似的,empty 函数用来创建一个指定大小的、没有具体值的数组。

案例

这里我们创建一个 3×2×3 的三维空数组:

np.empty((,,))
Out:array([[[, , ],[, , ]],   [[, , ],[, , ]],   [[, , ],[, , ]]])

这里要注意一下 empty 函数的输出。因为 empty 考虑到输出全 0 数组是不安全的想法,因此很多情况下,它返回的是一些没有规律的垃圾值。

Numpy 还提供了创建单位方阵(对角线为 1,其余为 0)的方法:identity 和 eye。

3.4 numpy.identity 函数

identity 函数用来创建一个指定大小的单位矩阵(对角线为 1 的数组)。

案例

这里我们利用 identity 函数来创建一个大小为 3×3 的单位矩阵:

np.identity()
Out:array([[., ., .],   [., ., .],   [., ., .]])

3.5 numpy.eye 函数

eye 函数用来创建一个指定大小的、对角线为 1 的数组。

eye 函数是 identity 的升级版本。eye 函数的原型如下:

np.eye(N, M=None, k=, dtype=<type ‘float’>)

当 k 为默认值时,eye 函数可以创建大小为 N×M,主对角线为 1 的矩阵。

案例

创建一个大小为 3×4 的主对角线为 1 的数组。

np.eye(N=, M=)
Out:array([[., ., ., .],   [., ., ., .],   [., ., ., .]])

案例

当 k、M 均为默认值时,eye 函数的功能和 identity 一致。

例如,利用 eye 函数创建一个大小为 3×3 的单位矩阵。

np.eye(N=)
Out:array([[., ., .],   [., ., .],   [., ., .]])

k 表示为 1 的对角线的偏移量,当 k 大于 0 时,表示向右偏移。

例如,创建一个大小为 3×4 的数组,为 1 的对角线向右偏移一位:

np.eye(N=, M=, k=)
Out:array([[., ., ., .],   [., ., ., .],   [., ., ., .]])

4. 小结

本小节重点讲述了如何利用序列型对象,创建一个数组序列;以及如何创建常用的特殊数组,例如:全 0 数组、全 1 数组,全空数组、对角线为 1 的数组等。