Добавить массив numpy в панду dataframa

Question

Я пытаюсь добавить массив моих оценок прогнозирования с ключом «прогноз» в существующий фрейм данных и сохранить его в файле 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

, если фрейм данных, который вы хотите добавить пред вектора, называется df; попробуйте использовать:

df['prediction'] = pred

он создаст столбец в df с именем «Предсказание» и присвоит этому столбцу значения вектора pred.