如何从 numpy 数组中提取每第 n 行

Question

我有一个 numpy 数组，我想从中提取每 3 行

输入

0.00    1.0000 
0.34    1.0000 
0.68    1.0000 
1.01    1.0000
1.35    1.0000
5.62    2.0000

我需要每 3 行提取一次，这样预期的 output 将是

0.68    1.0000 
5.62    2.0000

我的代码：

import numpy as np
a=np.loadtxt('input.txt')
out=a[::3]

但它给出了不同的结果。希望专家指导我。谢谢。

Answer 1

未定义时，（正）切片的起点是第一项。

您需要从第 n-1 项开始切片：

N = 3
out = a[N-1::N]

Output：

array([[0.68, 1.  ],
       [5.62, 2.  ]])