Python 如何将三维数组保存到csv文件中？

我训练了一个有3个类的模型。现在我想把预测保存在csv文件中，它给我的错误数据必须是一维的

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)

因此，对于三维，解决方法将是：

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

和

希望能有帮助

---

谢谢

您可以使用

pandas

和

numpy

，它可以通过重塑阵列并将其保存到csv中来解决您的问题

像这样的二维：

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

因此，对于三维，解决方法将是：

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

和

希望能有帮助

谢谢