如何将火炬张量转换为熊猫数据框？答案

【问题标题】：How to convert torch tensor to pandas dataframe?如何将火炬张量转换为熊猫数据框？
【发布时间】：2021-04-14 17:54:04
【问题描述】：

我想将 Torch 张量转换为 pandas 数据帧，但通过使用 pd.DataFrame，我得到了一个填充了张量而不是数值的数据帧。

import torch
import pandas as  pd
x = torch.rand(4,4)
px = pd.DataFrame(x)

这是我在变量资源管理器中单击 px 时得到的结果：

0   1   2   3
tensor(0.3880)  tensor(0.4598)  tensor(0.4239)  tensor(0.7376)
tensor(0.4174)  tensor(0.9581)  tensor(0.0987)  tensor(0.6359)
tensor(0.6199)  tensor(0.8235)  tensor(0.9947)  tensor(0.9679)
tensor(0.7164)  tensor(0.9270)  tensor(0.7853)  tensor(0.6921)

【问题讨论】：

标签： python pandas pytorch

【解决方案1】：

我找到了一种可能的方法，即先将 torch 转换为 numpy：

import torch
import pandas as  pd

x = torch.rand(4,4)
px = pd.DataFrame(x.numpy())

【讨论】：

这比迪欣的回答要快

【解决方案2】：

您可以使用astype更改类型

px = pd.DataFrame(x).astype("float")
px
          0         1         2         3
0  0.847408  0.714524  0.286006  0.165475
1  0.136359  0.384073  0.398055  0.437550
2  0.843704  0.301536  0.576983  0.231726
3  0.293576  0.075563  0.811282  0.881705

【讨论】：

不错！谢谢。