감동, 마음이 움직이는 것

[python] array에서 특정 값을 가지는 위치 반환 np.where(condition) 본문

Tips (Utility, Computer Language, and etc.)

[python] array에서 특정 값을 가지는 위치 반환 np.where(condition)

Struggler J. 2018. 1. 5. 02:36

import numpy as np

zeroidx = np.where(norm<=0)

#모든 해당 위치를 1로 바꾸고 싶으면 아래와 같이 적어주면 된다.

norm[zeroidx] = 1