인프런 커뮤니티 질문&답변

df.loc[:, ['컬럼명', '컬럼명']]

안녕하세요, 인프런 AI 인턴입니다.

wsyang님의 질문에 답변드리겠습니다.

iloc[]를 사용하여 떨어진 두 개의 피처(열)을 선택하는 방법에 대해 궁금해하셨네요. iloc[]에 원하는 열의 인덱스를 리스트로 전달하면 됩니다. 예를 들어 age와 income이라는 두 개의 피처가 있고, 각각의 인덱스가 2와 4라고 가정할 때, 다음과 같이 작성할 수 있습니다:

# 데이터프레임이 df일 때
selected_features = df.iloc[:, [2, 4]]

이렇게 하면 age와 income 열이 선택되어 분리될 것입니다.

관련하여, 기존에 유사한 방식으로 데이터프레임에서 특정 열을 선택하는 방법에 대한 더 많은 예제를 보시려면 다음의 게시물을 참고해 주세요:

1. Feature Engineering에서 인코딩 후 데이터를 합칠 때 iloc를 활용하셨는데
   
2. 원핫 인코딩 후 분리
   

이 게시물들은 iloc를 활용하여 데이터를 처리하는 방법을 구체적으로 설명하고 답변자께서도 범용적인 방법들을 제시하고 있습니다.

저는 질문자님의 이해를 돕고자 지식공유자님의 기존 답변을 참고하여 유사한 답변 링크를 추천해드리고 있습니다. 현재 베타 기능으로 답변이 만족스럽지 않을 수 있는 점 양해 부탁드립니다. 🙏 추가적으로 궁금한 점이 있으시면, 이어서 질문해 주세요. 곧 지식공유자께서 답변해 주실 것입니다.