Python numpy ndarray.getfield用法及代码示例

本文简要介绍 python 语言中 numpy.ndarray.getfield 的用法。

用法:

ndarray.getfield(dtype, offset=0)

以特定类型返回给定数组的字段。

字段是具有给定数据类型的数组数据的视图。视图中的值由给定类型和当前数组的偏移量(以字节为单位)确定。偏移量需要使视图 dtype 适合数组 dtype；例如，一个 dtype complex128 数组有 16 字节的元素。如果使用 32 位整数(4 个字节)进行查看，则偏移量需要在 0 到 12 个字节之间。

参数：

dtype： str 或 dtype

视图的数据类型。视图的 dtype 大小不能大于数组本身的大小。

offset： int

在开始元素视图之前要跳过的字节数。

例子：

>>> x = np.diag([1.+1.j]*2)
>>> x[1, 1] = 2 + 4.j
>>> x
array([[1.+1.j,  0.+0.j],
       [0.+0.j,  2.+4.j]])
>>> x.getfield(np.float64)
array([[1.,  0.],
       [0.,  2.]])

通过选择 8 个字节的偏移量，我们可以为我们的视图选择数组的复杂部分：

>>> x.getfield(np.float64, offset=8)
array([[1.,  0.],
       [0.,  4.]])