루프를 사용하지 않고 아래 작업을 수행 할 수있는 방법이 있습니까?
A = np.array([[1,1],[2,2],[3,3],[4,4],[5,5]])
I = np.array([0, 1, 0, 1, 1])
V = np.array([6, 6, 6, 6, 6])
I가 지정한 해당 인덱스에 6이 포함되도록 A를 업데이트하고 싶습니다. 따라서 A는 ...
A = np.array([[6,1],[2,6],[6,3],[4,6],[5,6]])
다음을 시도했지만 작동하지 않았습니다 ..
A[I] = V
A[:,I] = V
시도 : A[np.arange(len(A)), I] = V
:
In [15]: import numpy as np
In [16]: A = np.array([[1,1],[2,2],[3,3],[4,4],[5,5]])
In [17]: I = np.array([0, 1, 0, 1, 1])
In [18]: V = np.array([6, 6, 6, 6, 6])
In [19]: A[np.arange(len(A)), I] = V
In [20]: A
Out[20]:
array([[6, 1],
[2, 6],
[6, 3],
[4, 6],
[5, 6]])
이 기사는 인터넷에서 수집됩니다. 재 인쇄 할 때 출처를 알려주십시오.
침해가 발생한 경우 연락 주시기 바랍니다[email protected] 삭제
몇 마디 만하겠습니다