时间戳

首页 > 解决方案 > 如何将 3 维数组保存到 csv 文件中？

python - 如何将 3 维数组保存到 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]]

标签： pythonpandasdataframe

解决方案

您可以使用pandasand 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

希望能帮助到你

谢谢

推荐阅读