Set numpy array elements to zero if they are above a specific threshold

前端 未结 3 1370
南笙
南笙 2020-12-12 21:11

Say, I have a numpy array consists of 10 elements, for example:

a = np.array([2, 23, 15, 7, 9, 11, 17, 19, 5, 3])

Now I want to eff

3条回答
  •  离开以前
    2020-12-12 22:00

    If you don't want to change your original array

    In [1]: import numpy as np
    
    
    In [2]: a = np.array([2, 23, 15, 7, 9, 11, 17, 19, 5, 3])
    
    
    In [3]: b = np.where(a > 10, 0, a)
    
    
    In [4]: a
    
    Out[4]: array([ 2, 23, 15,  7,  9, 11, 17, 19,  5,  3])
    
    
    In [5]: b
    
    Out[5]: array([2, 0, 0, 7, 9, 0, 0, 0, 5, 3])
    

提交回复
热议问题