특정 numpy 행렬 열을 업데이트하십시오.

Question

다음 대상 색인에 해당하는 색인 ​​[2 4 3 4]이 있습니다. 다음 코드 줄 targets = np.zeros((features.shape[0], 5))을 사용하여 0의 행렬을 만듭니다. 임은 내가 한 번에 모두 특정 인덱스를 업데이트하고 for 루프없이 1에 해당 값을 설정할 수있는 방법으로 슬라이스 할 수는 이상적으로 매트릭스
([0,0,1,0,0], [0,0,0,0,1], [0,0,0,1,0], [0,0,0,0,1])

Answer 1

처럼 보일 것인지 궁금 나는 당신이 할 수 있다고 생각합니다 이 같은 :

array([[ 0., 0., 1., 0., 0.], 
     [ 0., 0., 0., 0., 1.], 
     [ 0., 0., 0., 1., 0.], 
     [ 0., 0., 0., 0., 1.]]) 

: 여기

targets = np.zeros((4, 5)) 
ind = [2, 4, 3, 4] 
targets[np.arange(0, 4), ind] = 1 

은 결과입니다