首先介绍一些有用的数组属性。定义三个随机的数组:一个一维数组、一个二维数组和一个三维数组:
import numpy as np
x1 = np.random.randint(10, size=6) # 一维数组
x2 = np.random.randint(10, size=(3, 4)) # 二维数组
x3 = np.random.randint(10, size=(3, 4, 5)) # 三维数组
每个数组有 nidm(数组的维度)、shape(数组每个维度的大小)和size(数组的总大小)属性:
print("x3 ndim: ", x3.ndim)
print("x3 shape:", x3.shape)
print("x3 size: ", x3.size)
得到输出结果如下:
x3 ndim: 3
x3 shape: (3, 4, 5)
x3 size: 60
另外一个有用的属性是 dtype,它是数组的数据类型:
print("dtype:", x3.dtype)
得到输出结果如下:
dtype: int64
其他的属性包括表示每个数组元素字节大小的 itemsize,以及表示数组总字节大小的属性 nbytes:
print("itemsize:", x3.itemsize, "bytes")
print("nbytes:", x3.nbytes, "bytes")
得到输出结果如下:
itemsize: 8 bytes
nbytes: 480 bytes
一般来说,可以认为 nbytes 跟 itemsize 和 size 的乘积大小相等。