在python中拆分数组

Question

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

Answer 1

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

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

Answer 2

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]