Python 获取一维np数组的片段

我对python和机器学习非常陌生

假设我有一个1D np数组（包含数字和NaN），其中有一列和1308行，并希望创建两个变量：

训练结果=结果[0:891,0]
y_pred=结果[891:，0]

我试过了，结果很明显

---

我是如此绝望，我把它转换回DF进行手术。必须有一种更简单的方法来实现这一点。

如果数组有一维，则不需要逗号。以下是我的做法：

train_outcome = outcome[:891]
y_pred = outcome[891:]

---

如果数组有一个维度，则不需要逗号。以下是我的做法：

train_outcome = outcome[:891]
y_pred = outcome[891:]

---

使用

np.split

作为一个班轮：

train_outcome, y_pred = np.split(outcome, [891])

---

使用

np.split

作为一个班轮：

train_outcome, y_pred = np.split(outcome, [891])