0이 아닌 경우 ndarray의 모든 요소에 1을 추가하십시오.

Question

여기 Python noobie :) 사물을 간결하게 처리하는 방법을 찾으려합니다.

>>> f = np.zeros((2,2), dtype=int) 
>>> f[0][1] = 100 
>>> f[1][0] = 200 
>>> print f 

[[ 0 100] 
[200 0]] 

>>> f1 = # do something special 
>>> print f1 

[[ 0 101] 
[201 0]] 

고마워요!

Answer 1

약간 더 나은 방법 boolean indexing을 사용하는 것이다

>>> import numpy as np 
>>> a = np.array([[0,100],[200,0]]) 
>>> a[a!=0] += 1 
>>> a 
array([[ 0, 101], 
     [201, 0]]) 

오히려 잘 find가 np.where와 유사한 역할을 제공하는 매트랩 포럼에 설명되는 개선 이유. 예를 들어, this Matlab thread에서는 find이 부울 배열로 호출됩니다. find 다음 인덱스를 추출하는 몇 가지 추가 기능을 수행합니다. 이 색인은 numpy에서 "멋진 색인 생성"으로 알려진 것을 수행하는 데 사용됩니다. 그러나 배열 자체는 부울 배열로 멋진 방식으로 색인을 생성 할 수있었습니다.

Answer 2

numpy.where을 사용하여 값이 0이 아닌 부분을 파악한 다음 모든 값에 1을 더할 수 있습니다.

>>> a = np.array([[0,100],[200,0]]) 
>>> a 
array([[ 0, 100], 
     [200, 0]]) 

>>> a[np.where(a != 0)] += 1 
>>> a 
array([[ 0, 101], 
     [201, 0]])