获取一维 np 数组的片段

Question

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

假设我有一个一列 1308 行的一维 np 数组（同时包含数字和 NaN），并且想要创建两个变量：

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

我尝试了这个并得到了明显的 <IndexError: too many indices for the array: array is 1-dimensional, but 2 was indexed>。

我非常绝望，以至于将其转换回 DF 进行操作。 必须有一种更简单的方法来实现这一点。

Answer 1

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

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

Answer 2

将np.split用于单行：

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