문돌이 존버/데이터 분석
2021. 6. 16.
판다스 iloc로 여러 컬럼 선택하기 feat. np.r_
판다스 데이터프레임에서 여러 컬럼을 가져오고 싶을 때 어떻게 해야 할까요? loc 메서드를 쓰면 컬럼 이름을 직접 입력하면 되고, iloc 메서드를 쓰면 컬럼 인덱스를 사용해면 되겠죠. 그런데 만약 iloc를 사용해서 컬럼 인덱스를 가져올 때 연속된 인덱스가 아니라 서로 떨어져 있는 인덱스의 경우 곤란해집니다. 즉 0~3까지 인덱스에 10번째 인덱스가 필요한 상황인 것이죠. 이때 사용할 수 있는 방법이 np.r_ 입니다. df.iloc[:, np.r_[:3, 10]] 위의 코드는 데이터프레임의 모든 행과 0, 1, 2, 10 인덱스에 해당하는 열을 가져오는 것입니다. 이해를 돕기 위해 np.r_ 만 단독으로 사용하였을 경우 결과값을 첨부했습니다. np.r_[:3, 10] np.r_에 파라미터로는 총 3개..
