numpy 带字节字段的结构化数组

Question

我正在尝试构建一个结构化的 numpy 数组，其中的字段是命名字节字段（类型4b ，不是 unicode）。

import numpy as np 
dtype = np.dtype([('count', 'u8'), ('name', '4b')], align=True)
a = np.asarray([(10, b'test')], dtype=dtype)
print(a.dtype)

我收到错误：

ValueError: invalid literal for int() with base 10: b'test'

现在，如果我将字节字段更改为 unicode，

import numpy as np 
dtype = np.dtype([('count', 'u8'), ('name', 'U4')], align=True)
a = np.asarray([(10, 'test')], dtype=dtype)
print(a.dtype)

这不会导致错误，我得到 output：

{'names':['count','name'], 'formats':['<u8','<U4'], 'offsets':[0,8], 'itemsize':24, 'aligned':True}

但对我来说，这是一个 hack，因为我特别想要字节。

问题：如何在我的结构化 numpy 数组上获取命名字节字段？

Answer 1

您可以使用S类型字符串来处理字节字符串：

>>> dtype = np.dtype([('count', 'u8'), ('name', 'S4')], align=True)
>>> a = np.array([(10, b'test')], dtype=dtype)
>>> a
array([(10, b'test')],
      dtype={'names':['count','name'], 'formats':['<u8','S4'], 'offsets':[0,8], 'itemsize':16, 'aligned':True})