numpy 在tf-idf数组上进行预测,如何与原始 Dataframe 合并

klr1opcd  于 12个月前  发布在  其他
关注(0)|答案(1)|浏览(120)

我有文本数据,Pandas Dataframe ,其中,我转换成tf-idf向量比使用ML算法预测done.so,如何合并xtest的索引与xtestPandas Dataframe 的结果.我已经执行了train_test_split后,对词嵌入,而不是对Pandas Dataframe .当我检查:

array.index()
Numpy array has no attribute index

字符串
数组的形状是(3480,5000),当我尝试下面的代码:

len(array.indices)
10480

iq0todco

iq0todco1#

假设pred是你的预测
它是numpy数组,然后你可以df。将preds作为Series合并到main df

import pandas as pd
import numpy as np

data = {'A': [1, 2, 3, 4], 'B': [5, 6, 7, 8]}
df = pd.DataFrame(data, index=[1, 2, 3, 4])

pred = np.array([0.1, 0.9, 0.5, 0.11])

series = pd.Series(pred, index=df.index, name='C')
result = df.merge(series, left_index=True, right_index=True)

print(result)

字符串

相关问题