如何从numpy数组中每隔n行抽取

zqdjd7g9 于 2023-02-12 发布在其他

关注(0)|答案(1)|浏览(222)

我有一个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行提取一次，以便预期输出为

0.68    1.0000 
5.62    2.0000

我的代码：

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

但是结果不一样，希望Maven指导。谢谢。

1条答案

未定义时，（正）切片的起点是第一个项目。
您需要从第n-1项开始切片：

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

输出：

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