python - 根据信息拆分数组

Question

我正在用 Python 做一个项目，其中包含 600 个小行星测量值(直径、周期、轨道半径)的数组。有了这些信息，我想根据直径列制作新的阵列，得到的阵列将是第一个阵列的直径间隔。

如果我有这个数组，

diameter           period         orbit radius
 [[  19.17          5.71476129    3.19639121]
 [  19.28          4.81234455    2.85035536]
 [  22.77          5.62890294    3.16429553]
 [   4.8           3.29145453    2.21268592]
 [   7.23          2.61331495    1.89724041]
 [   8.17          2.54935585    1.86615697]
 [ 260.94          6.49151957    3.47983602]
 [ 530.            3.62867648    2.3613482 ]
 [ 952.4           4.60562864    2.76813421]]

我想做一个直径从0到20的数组

diameter           period         orbit radius
 [[  19.17          5.71476129    3.19639121]
 [  19.28          4.81234455    2.85035536]
 [   4.8           3.29145453    2.21268592]
 [   7.23          2.61331495    1.89724041]
 [   8.17          2.54935585    1.86615697]]

Answer 1

只需使用列表理解来过滤它

filtered = [ ast for ast in arr if ast[0]<=20 ]