当数组第三列的元素小于特定数量时,我想删除数组的行。例如:
a=np.array([[2331.13,1944.88,23.1379,7,3.18339,0.482105], [8168.44,1904.70,19.5025,265,4.12642,0.0376510], [7389.36,1983.97,14.3581,3937,6.04109,0.713416], [1765.18,1944.29,22.5495,35,2.30717,0.794432], [2319.33,1946.68,22.4300,25,3.63676,0.0210690], [785.666,2090.69,14.7940,1095,2.52823,0.999842], [4071.24,2186.92,22.6616,31,2.79309,0.0312501], [7082.51,2191.69,23.0122,19,2.53166,0.687001]])
我想删除满足以下条件的行:
a[:,2]<15.0
干杯。
你可以做:
a[a[:,2]>=15.0, :]
注意a[:,2]<15.0to的倒置a[:,2]>=15.0,以便您描述要 保留 而不是 删除的行 。
a[:,2]>=15.0
如果反转条件不是那么简单,则还可以使用~:
~
a[~(a[:,2]<15.0), :]