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. ]]