在pandas DataFrame中附加pytorch对象

Question

我想在空数据帧df_data1_cluster 中附加张量对象c，s

df_data1_cluster = pd.DataFrame(columns = ["cluster", "text"])
label, center = detect_clusters(torch.as_tensor(embeddings), 50)
for c, s in zip(label, phrases):
df_data1_cluster.append(c,s)

这导致了错误。

TypeError: cannot concatenate object of type '<class 'torch.Tensor'>'; only Series and DataFrame objs are valid

Answer 1

Hi这是panda问题，append命令需要u附加一个数据帧，有一些方法可以插入到数据帧

import pandas as pd
import torch
df = pd.DataFrame({'numbers': [1, 2, 3], 'colors': ['red', 'white', 'blue']})
df['numbers'].loc[-1] = torch.tensor([2, 3, 4])  # adding a row