Как сохранить трехмерный массив в CSV-файл?

Question

Я тренировал модель, у которой есть 3 числа классов. Теперь я хочу сохранить прогноз в CSV-файле, который выдает мне ошибку, данные должны быть 1-мерными.

prediction1= loaded_model.predict(X_test,verbose=1)
prediction1 = pd.DataFrame({
    "phrase": test["Phrase"],
    "predict":prediction1
})
[output] Erorr
Data must be 1-dimensional

print(prediction1)
[[0.2634249  0.27810773 0.45846733]
[0.2634249  0.27810773 0.45846733]
[0.2634249  0.27810773 0.45846733]
...
[0.2634249  0.27810773 0.45846733]
[0.2634249  0.27810773 0.45846733]
[0.2634249  0.27810773 0.45846733]]

Answer 1

Вы можете использовать pandas и numpy, это может решить вашу проблему путем изменения формы массива и сохранения его в csv

Примерно так: 2 Dimensional:

import numpy as np
x = np.arange(10).reshape((2,5))
np.savetxt('save.txt', x)

Итак,для 3-х мерного обходного пути будет:

import numpy as np
import pandas as pd    
x = np.arange(60).reshape((2,5,6))
with file('test.txt', 'w') as outfile:
    for slice_2d in x:
        np.savetxt(outfile, slice_2d)

Вы также можете использовать pickle и numpy.ndarray.tofile

Надеюсь, это поможет

Спасибо