1

我有以下代码:

import numpy as np
a=np.array([[1.1,5,100],[2.4,6,200],[3.3,7,300],[4.1,8,400],[4.9,9,500],[5.5,10,600],[6,5],700,[6.5,12,800],[7.2,20,900],[8,20,1000]])
b=np.max(a[:,0])#finds maximum value in 1st column(index=0)
if b > 5:
   Do something

我希望代码识别数组第 1 列中值大于 5 的第一行,然后删除之后的行。仅当第 1 列中的最大值大于 5 时才会发生这种情况。结果数组 c 应如下所示:

c=np.array([[1.1,5,100],[2.4,6,200],[3.3,7,300],[4.1,8,400],[4.9,9,500]])
4

1 回答 1

3

找到的解决方案:

import numpy as np
a=np.array([[1.1,5,100],[2.4,6,200],[3.3,7,300],[4.1,8,400],[4.9,9,500],[5.5,10,600],[6,5],700,[6.5,12,800],[7.2,20,900],[8,20,1000]])
b=np.max(a[:,0])#finds maximum value in 1st column(index=0)
if b > 5:
    c=a[a[:,0]<=5,:]

我在以下位置找到了解决方案:

如果行的元素之一不满足条件,我如何删除数组的行?

于 2013-10-18T12:41:06.970 回答