Numpy에서 특정 조건의 열을 추출하고 싶습니다.

조회수 81회

Numpy에서 특정 조건의 열을 추출하고 싶습니다.

데이터 셋에 Pandas를 이용하여 Numpy 배열을 얻었습니다.

[[0.2  0.   0.   ... 0.   2.   2.  ]
 [0.2  0.   0.   ... 0.   2.   2.  ]
 [0.55 0.   0.   ... 0.   0.   0.  ]
 [0.43 0.   0.   ... 0.   2.   2.  ]
 [0.21 0.   0.   ... 0.   2.   1.  ]
 [0.37 0.   0.   ... 0.   2.   2.  ]]

여기에서 각 열의 마지막 element (0,1,2)를 기준으로 파일을 수식을 이용해서 재분류 하고 싶은데 어떻게 해야 되나요?

ex)

print(train0) 
 [ [0.55 0.   0.   ... 0.   0.   0.  ]]

print(train1) 
 [ [0.21 0.   0.   ... 0.   2.   1.  ]]

print(train2) 
 [[0.2  0.   0.   ... 0.   2.   2.  ]
 [0.2  0.   0.   ... 0.   2.   2.  ]
 [0.43 0.   0.   ... 0.   2.   2.  ]
 [0.37 0.   0.   ... 0.   2.   2.  ]]

1 답변

답변을 하려면 로그인이 필요합니다.

Hashcode는 개발자들을 위한 무료 QnA 사이트입니다. 계정을 생성하셔야만 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)

ᕕ( ᐛ )ᕗ
로그인이 필요합니다

Hashcode는 개발자들을 위한 무료 QnA사이트 입니다. 계정을 생성하셔야만 글을 작성하실 수 있습니다.