我有以下代码:
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]])