在python中拆分数组

【问题标题】：Splitting a array in python在python中拆分数组
【发布时间】：2018-04-05 04:30:52
【问题描述】：

如何根据数组中元素的数量在python中拆分数组。我在做 knn 分类，我需要考虑 2D 数组的前 k 个元素。

【问题讨论】：

【解决方案1】：

import numpy as np
x = np.array([1, 2, 4, 4, 6, 7])

print(x[range(0, 4)])

您还可以通过获取要使用的元素范围来拆分它。您可以将 x[range(x, x)]) 存储在变量中，也可以使用数组的那些特定元素。如您所见，输出将数组拆分：

[1 2 4 4]

【讨论】：

【解决方案2】：

在Numpy中，有一个方法numpy.split.

x = np.arange(9.0)
np.split(x, 3)

【讨论】：