使用numpy库在python中拆分数组

Question

我是python初学者，想知道np.split(x, [3, 5])中[3,5]的用途

x = [1, 2, 3, 99, 99, 3, 2, 1]
x1, x2, x3 = np.split(x, [3, 5])
print(x1, x2, x3)

输出：

[1 2 3] [99 99] [3 2 1]

Answer 1

[3,5] 在索引 3 和 5 处拆分 numpy 数组。在索引 3 处有 99，在索引 5 处有 3，因此它已根据这些值拆分数组。

Answer 2

出现此类问题时，请先阅读文档。 Here 你可以看到numpy.split 有(ary, indices_or_sections, axis=0) 参数。您将x 作为第一个参数ary 和[3, 5] 作为indices_or_sections 传递。如果你看indices_or_sections参数解释你会注意到：

如果 indices_or_sections 是排序整数的一维数组，则条目 指示数组沿轴拆分的位置。

例如，[2, 3] 对于axis=0，会导致ary[:2] ary[2:3] ary[3:]

所以您的代码只是将x 拆分为沿零轴的第三个和第五个索引。