import sys
import numpy as np
a=np.array([4,2,3,1])
remove_ixs=np.where(a>1)[0]
b=np.delete(a,remove_ixs)
print(b)
输出:
[1]
import sys
import numpy as np
a=np.array([4,2,3,1])
remove_ixs=np.where(a>1)[0]
b=np.delete(a,remove_ixs)
print(b)
输出:
[1]