python array끼리의 나눗셈에서 분모가 0일 때 NAN으로 만들기
조회수 763회
PER = EndPrice / EPS 이라는 파이썬 식에서
EndPrice, EPS는 각각 2차원 array인데
EPS가 0인 경우가 있어서 Error 가 낪니다.
EPS가 0인경우에는 NAN 값을 반환하고 싶은데
어떻게 코드를 짜면 될까요?
2 답변
-
-
numpy array 를 다루는 걸로 보입니다. 이럴 때에는 아래와 같은 코드가 가능하군요.
>>> a = np.array([1,2,3]) >>> b = np.array([1,4,0]) >>> a/b Warning (from warnings module): File "<pyshell#21>", line 1 RuntimeWarning: divide by zero encountered in true_divide array([1. , 0.5, inf]) >>> c = a/b >>> c array([1. , 0.5, inf]) >>> c[c==np.inf] array([inf]) >>> c[c==np.inf] = np.nan >>> c array([1. , 0.5, nan])
댓글 입력