Python中使用msgpack_numpy编解码时如何保留字符串字典键？

这个问题的核心在于encoding='utf-8'参数会干扰msgpack_numpy的解码逻辑——它会把msgpack_numpy用来识别numpy数组的内部字节标记也当成普通字符串解码，导致数组被拆成了原始字典结构。这里有两个靠谱的解决方案：

方案一：手动转换字典键（简单直接，兼容你的版本） #

解码时不指定encoding参数，先得到带字节类型键的字典，再手动将键解码为UTF-8字符串。这种方法不会影响numpy数组的正常解码：

import numpy as np
import msgpack
import msgpack_numpy as m

m.patch()

# 打包包含numpy数组的字典
d = {'key': np.arange(5)}
binary = msgpack.packb(d)

# 基础解码（保留numpy数组，键为bytes类型）
ret = msgpack.unpackb(binary)
# 将bytes键转换为字符串
ret_str_keys = {k.decode('utf-8'): v for k, v in ret.items()}

print(ret_str_keys.keys())  # 输出: dict_keys(['key'])
print(ret_str_keys['key'])  # 输出: array([0, 1, 2, 3, 4])

方案二：使用raw=False参数（更优雅，适配msgpack 0.5+） #

在Python 3环境下，msgpack的raw=False参数会自动将原本是字符串类型的字节序列解码为UTF-8字符串，同时不会破坏msgpack_numpy识别numpy数组的标记（因为这些标记属于专用二进制结构，不会被raw=False处理）：

import numpy as np
import msgpack
import msgpack_numpy as m

m.patch()

d = {'key': np.arange(5)}
binary = msgpack.packb(d)

# 用raw=False解码，直接得到字符串键+正常numpy数组
ret = msgpack.unpackb(binary, raw=False)

print(ret.keys())  # 输出: dict_keys(['key'])
print(ret['key'])  # 输出: array([0, 1, 2, 3, 4])

为什么encoding='utf-8'不行？因为这个参数会强制解码所有字节序列，包括msgpack_numpy用来标识numpy数组的内部元数据，导致这些元数据被破坏，无法正确还原为ndarray对象。而raw=False只会处理那些原本就是字符串的字节（比如你的字典键），完美避开了msgpack_numpy的专用结构。

内容的提问来源于stack exchange，提问作者Liran Funaro