python - 'numpy.ndarray' 对象如何不 'numpy.ndarray' 对象？

Question

当您调用 DataFrame.to_numpy() 时，pandas 将找到可以容纳 DataFrame 中所有数据类型的 NumPy 数据类型。但是如何进行反向操作呢？

我有一个“numpy.ndarray”对象“pred”。它看起来像这样:

[[0.00599913 0.00506044 0.00508315 ... 0.00540191 0.00542058 0.00542058]]

我正在尝试这样做:

 pred = np.uint8(pred)
 print("Model predict:\n", pred.T)

但我得到:

[[0 0 0 ... 0 0 0]]

为什么，转换后，我没有得到这样的东西:

0 0 0 0 0 0 ... 0 0 0 0 0 0

以及如何将 pred 写入文件？

pred.to_csv('pred.csv', header=None, index=False)
pred = pd.read_csv('pred.csv', sep=',', header=None)

给出错误信息:

AttributeError                            Traceback (most recent call last)
<ipython-input-68-b223b39b5db1> in <module>()
----> 1 pred.to_csv('pred.csv', header=None, index=False)
      2 pred = pd.read_csv('pred.csv', sep=',', header=None)
AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'

请帮我解决这个问题。

Answer 1

您可以使用一行代码将 ndarray 转换为 pandas df，然后转换为 csv 文件来解决该问题。

pd.DataFrame(X_train_res).to_csv("x_train_smote_oversample.csv")