浮点数组到字节数组和反之亦然

Posted 2021-04-14

我的最终目标是能够通过UDP套接字发送一个浮点数组，但是现在我只想尝试在python3中运行一些东西。下面的代码工作得很好：

import struct
fake_data = struct.pack('f', 5.38976) 
print(fake_data) 
data1 = struct.unpack('f', fake_data) 
print(data1)

输出：

 b'xeaxxac@'
(5.3897600173950195,)

但是当我尝试这个时，我得到：

electrode_data = [1.22, -2.33, 3.44]

for i in range(3):
    data = struct.pack('!d', electrode_data[i])  # float -> bytes
    print(data[i])
    x = struct.unpack('!d', data[i])  # bytes -> float
    print(x[i])

输出：

63
Traceback (most recent call last):
File "cbutton.py", line 18, in <module>
x = struct.unpack('!d', data[i])  # bytes -> float
TypeError: a bytes-like object is required, not 'int'

如何将float数组转换为字节数组，反之亦然。我试图完成这个的原因是因为第一个代码允许我使用UDP套接字将浮动数据从客户端发送到服务器（逐个）。我的最终目标是使用数组执行此操作，以便使用matplotlib绘制值。

答案

你这里只包装一个浮子。但是，你试图将结果缓冲区的第一个字节（隐式转换为int）传递给unpack。你需要给它整个缓冲区。此外，要以更一般的方式执行此操作，您需要首先将数组中的项目数编码为整数。

import struct

electrode_data = [1.22, -2.33, 3.44]
# First encode the number of data items, then the actual items
data = struct.pack("!I" + "d" * len(electrode_data), len(electrode_data), *electrode_data)
print(data)

# Pull the number of encoded items (Note a tuple is returned!)
elen = struct.unpack_from("!I", data)[0]
# Now pull the array of items
e2 = struct.unpack_from("!" + "d" * elen, data, 4)
print(e2)

（*electrode_data意味着压扁名单：它与electrode_data[0], electrode_data[1]...相同）

如果你真的只想一次做一个：

for elem in electrode_data:
    data = struct.pack("!d", elem)
    print(data)
    # Again note that unpack *always* returns a tuple (even if only one member)
    d2 = struct.unpack("!d", data)[0]
    print(d2)