numpy - 从 numpy 行向量中每第 5 和第 6 个元素切片

Question

我有一个这样的数组：A = ([0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16])并且想从这个行向量中提取每第 5 个“和”第 6 个元素，导致B = ([4,5,9,10,14,15]).

我知道如何提取每 5 个元素：B = A[::5]但不知道如何一个接一个地提取两个值。

标签： numpynumpy-slicing

Answer 1

您可以使用：

A[4::5]

得到array([ 4, 9, 14])然后连接和排序A[::5]。