Python 3中如何将字节数组一次性转换为浮点数组？

问题场景 #

你在C程序中输出了两个float的二进制数据：

float array[2] = { 1.0, 2.0 };
fwrite(&array[0], sizeof(array[0]), sizeof(array) / sizeof(array[0]), stdout);
fflush(stdout);

Python读取STDOUT后得到字节串：b'\x00\x00\x80?\x00\x00\x00@'

你尝试逐个解码是成功的：

import struct

data = b'\x00\x00\x80?\x00\x00\x00@'
struct.unpack('<f', data[0:4])  # 返回 (1.0,)
struct.unpack('<f', data[4:8])  # 返回 (2.0,)

但一次性解码整个字节串时失败：

struct.unpack('<f', data)

报错信息：

error: unpack requires a buffer of 4 bytes

解决方案：一次性解码完全可行！ #

问题出在格式字符串的写法上：'<f' 只会告诉struct.unpack去解码一个小端序的float（占4字节），但你的字节串是8字节，对应两个float，所以格式字符串需要匹配数据的数量。

你只需要把格式字符串改成'<2f'，就可以一次性解码两个float：

import struct

data = b'\x00\x00\x80?\x00\x00\x00@'
result = struct.unpack('<2f', data)
print(result)  # 输出: (1.0, 2.0)

如果你的字节串长度不确定（比如对应N个float），还可以动态生成格式字符串：

num_floats = len(data) // 4  # 每个float固定占4字节
result = struct.unpack(f'<{num_floats}f', data)

简单来说，格式字符串里的数字表示要解码的对应类型的数量，这样struct.unpack就会自动按照字节长度拆分缓冲区，一次性返回所有解码后的数值元组，完全不需要循环逐个处理~

内容的提问来源于stack exchange，提问作者Marko Gulin