如何将pandas DataFrame转换为字节，反之亦然？

Question

我正在尝试使用套接字编程发送数据帧。为此，我需要知道如何在客户端将数据帧编码为字节，然后将其解码为服务器上的数据帧。

Answer 1

您可以使用cPickle或pickle模块。

首先序列化数据帧：

>>> df
   A  B
0  1  3
1  2  4
>>> df_bytes = cPickle.dumps(df)

以某种方式发送df_bytes的内容，然后反序列化：

>>> df2 = cPickle.loads(df_bytes)
>>> df2
   A  B
0  1  3
1  2  4

这两个模块确实引入了潜在的安全问题，因为它们允许在反序列化期间执行任意代码。更好的解决方案是将数据帧写入json字符串：

df_string = df.to_json()

然后恢复它：

df_again = pandas.read_json(df_string)

其他格式如csv也可用，相应的功能为DataFrame.to_csv()和pandas.read_csv()。有关相关功能的完整列表，请参阅this和this。