使用numpy库在python中拆分数组

Question

我是python的初学者，我想知道[3,5]中np.split(x, [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参数说明，您会注意到：

  

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

     

例如，[2，3]   对于轴= 0，将导致ary [：2] ary [2：3] ary [3：]

因此，您的代码仅沿零轴在第三个索引和第五个索引上分割了x。