将numpy数组添加到pandas dataframa

Question

我尝试将带有预测键“ prediction”的预测得分数组添加到现有数据框中，并将其保存在JSON文件中。我没问题，但是什么也没发生。怎么了

df_pred = pd.DataFrame(pred, columns=["prediction"])

frames = [df_Origin, df_pred]
df_merged = pd.concat(frames)

mergedJSON = df_merged.to_json(employeeFolder + '/predictionData.json', orient='records')

Answer 1

如果要添加pred向量的数据框名为df；尝试使用：

df['prediction'] = pred

它将在df中创建一个名为“ prediction”的列，并将pred向量值分配给该列。