如何將火炬張量轉換為熊貓數據幀？

Question

我想將火炬張量轉換為pd.DataFrame數據幀，但是通過使用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)

Answer 1

我通過首先將 torch 轉換為 numpy 找到了一種可能的方法：

import torch
import pandas as  pd

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

Answer 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