dataframe - 将张量的数据框保存到文件中以便于加载和访问的方法？

Question

我有一个包含 2600 行的数据框，每行都有形状为 torch 张量(192,)。

如何将此数据框保存到文件中，以便当我再次加载它时，我仍然可以使用“类似字典”的方式访问它的内容？

保存 to_csv() 将张量转换为字符串，导致我需要解析的地方变得混乱。

标签： dataframetorch

Answer 1

事实证明，使用pandas.to_pickle()函数保留了数据帧格式，并且在加载回数据后我仍然可以方便地访问数据。