[Python] Numpy 배열 요소 변환

np.where()

• np.where(조건, True일 때 값, False일 때 값)

• np.where을 사용하면 편리하게 배열 요소를 변환할 수 있습니다.

• 또한, 벡터 연산을 하므로 for loop을 돌지 않아서 속도가 매우 빠릅니다.

    import numpy as np
  
    a = np.arange(-5, 5)
    # array([-5, -4, -3, -2, -1,  0,  1,  2,  3,  4])
  
    b = np.where(a < 0, 0, a)
    # 0보다 작은 수는 0으로 나머지는 그대로 변환되며
    # 원래의 배열 a는 변경되지 않고 그대로 있습니다.
    # array([0, 0, 0, 0, 0, 0, 1, 2, 3, 4])