Python numpy recarray.getfield用法及代碼示例

本文簡要介紹 python 語言中 numpy.recarray.getfield 的用法。

用法:

recarray.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.]])