2022年 11月 13日

Python——将小于0的数换成0的几种方式

 
  1.  
  2. >>> import numpy as np
  3.  
  4. >>> a = np.random.randint(-5, 5, (5, 5))
  5.  
  6. >>> a
  7.  
  8. array([[-4, -4, -5,  2,  1],
  9.  
  10.        [-1, -2, -1,  3,  3],
  11.  
  12.        [-1, -2,  3, -5,  3],
  13.  
  14.        [ 0, -3, -5,  1, -4],
  15.  
  16.        [ 0,  3,  1,  3, -4]])
  17.  
  18. # 方式一
  19.  
  20. >>> np.maximum(a, 0)
  21.  
  22. array([[0, 0, 0, 2, 1],
  23.  
  24.        [0, 0, 0, 3, 3],
  25.  
  26.        [0, 0, 3, 0, 3],
  27.  
  28.        [0, 0, 0, 1, 0],
  29.  
  30.        [0, 3, 1, 3, 0]])
  31.  
  32. # 方式二
  33.  
  34. >>> (a + abs(a)) / 2
  35.  
  36. array([[0, 0, 0, 2, 1],
  37.  
  38.        [0, 0, 0, 3, 3],
  39.  
  40.        [0, 0, 3, 0, 3],
  41.  
  42.        [0, 0, 0, 1, 0],
  43.  
  44.        [0, 3, 1, 3, 0]])
  45.  
  46. # 方式三
  47.  
  48. >>> b = a.copy()
  49.  
  50. >>> b[b < 0] = 0
  51.  
  52. >>> b
  53.  
  54. array([[0, 0, 0, 2, 1],
  55.  
  56.        [0, 0, 0, 3, 3],
  57.  
  58.        [0, 0, 3, 0, 3],
  59.  
  60.        [0, 0, 0, 1, 0],
  61.  
  62.        [0, 3, 1, 3, 0]])
  63.  
  64. # 方式四
  65.  
  66. >>> np.where(a > 0, a, 0)
  67.  
  68. array([[0, 0, 0, 2, 1],
  69.  
  70.        [0, 0, 0, 3, 3],
  71.  
  72.        [0, 0, 3, 0, 3],
  73.  
  74.        [0, 0, 0, 1, 0],
  75.  
  76.        [0, 3, 1, 3, 0]])