python - 将张量添加到数据框

Question

我有一个具有以下特征的张量：

<class 'torch.Tensor'>

torch.Size([600, 1])

我希望它成为数据框中的一列。

当我将张量按原样添加到 df 时，我得到以下行之一。

tensor([-0.1186])

我该如何摆脱tensor([])？这样每一行都只是一个数字。对于这一行，我只想-0.1186.

标签： pythonpandasdataframepytorchtensor

Answer 1

如果您squeeze的张量从大小(600,1)到大小(600)，您可以直接将其值作为新列添加到 DataFrame：

df['newcol'] = my_tensor.squeeze()