如何从NumPy数组提取每隔第3行？现有代码结果不符求指导

问题原因 #

你当前用的切片a[::3]是从索引0开始，以步长3提取行，得到的是索引0、3对应的行：

0.00    1.0000 
1.01    1.0000

这和你的期望不符——你需要的是第3行（对应索引2）、第6行（对应索引5）这类位置的行。

修正后的代码 #

把切片的起始索引设为2（对应原数组的第3行），步长保持3即可：

import numpy as np
a = np.loadtxt('input.txt')
out = a[2::3]
# 打印结果
print(out)
# 若要保存到文件，可添加：
# np.savetxt('output.txt', out, fmt='%.2f    %.4f')

验证结果 #

运行修正后的代码，提取的行完全符合你的期望：

0.68    1.0000 
5.62    2.0000

内容的提问来源于stack exchange，提问作者pro