python - 如何将3维数组保存到csv文件中？

我训练了一个模型，该模型具有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]]

最佳答案

您可以使用pandas和numpy，它可以通过重塑数组并将其保存到csv中来解决您的问题

像这样的二维：

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

希望能帮助到你

谢谢