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

업데이트 되어 불린으로 나옵니다

연산하면 true는 1 false는 0으로 동일합니다 🙂

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

원핫인코딩을 통해 생성된 컬럼의 데이터 타입이 bool로 나오는 경우가 종종 발생할 수 있습니다. 이는 사용하고 있는 Pandas 버전의 차이 또는 기본 설정 차이에 의해 발생할 수 있습니다. 일반적으로, 최신 버전의 Pandas에서는 get_dummies() 함수가 boolean 타입을 반환하는 경향이 있습니다.

해결방법과 고려사항

1. Pandas 버전 확인 및 수정:

   • 최신 버전의 Pandas에서 get_dummies() 사용 시 기본적으로 bool 타입으로 반환될 수 있습니다. 만약 int 타입을 원하신다면, dtype 인자를 사용하여 명시적으로 설정할 수 있습니다:
     

   train = pd.get_dummies(train, dtype=int)
   test = pd.get_dummies(test, dtype=int)
   

   이와 같이 하면 bool 대신 int 형식으로 반환됩니다.

2. 왜 bool 타입으로 나타나는가:

   • Boolean 타입은 메모리를 효율적으로 사용할 수 있으며, 통계적인 분석이나 머신러닝 알고리즘에서 실질적으로 사용하는데 문제가 없습니다. 하지만 원하는 타입으로 변환할 필요가 있는 경우 위와 같이 dtype 옵션을 사용할 수 있습니다.
     

3. bool 타입으로 사용해도 되나요?:

   • bool 타입은 머신러닝 모델에서 사용 가능하며, 수학적인 연산을 실행하는 데 있어서도 문제가 되지 않습니다. 만약 특정 알고리즘이나 환경에서 문제가 발생하는 경우에만 변경하여 사용하시면 됩니다.
     

관련 질문 및 추가 참고

• 원핫인코딩 출력값 True, False: 이 게시물에서는 원핫 인코딩 후 출력을 int로 변경하는 방법을 다루고 있어 추가적인 도움이 될 수 있습니다.
  

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