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

baek님의 프로필 이미지
baek

작성한 질문수

[퇴근후딴짓] 빅데이터 분석기사 실기 (작업형1,2,3)

피처 엔지니어링

list, dtypes 활용

해결된 질문

작성

·

167

0

  • 스케일링 작업할 컬럼명을 list 로 활용할 때 질문이 있습니다.

  • 1개 데이터 타입은 아래처럼 쓰면 되는데,

  • cols = list(X_train.columns[X_train.dtypes == object])

  • int, float 를 list 로 한번에 활용하려면 어떻게 해야하는지 문의드립니다.

답변 1

0

퇴근후딴짓님의 프로필 이미지
퇴근후딴짓
지식공유자

import pandas as pd

# 예시 데이터프레임 생성
data = {
    'int_column': [1, 2, 3],
    'float_column': [1.1, 2.2, 3.3],
    'str_column': ['a', 'b', 'c']
}
df = pd.DataFrame(data)

# int와 float 타입 컬럼 선택
numeric_cols = list(df.select_dtypes(include=['int', 'float']).columns)

print(numeric_cols)

 

이 방법을 사용하거나 exclude = 'object'를 활용하는 방법이 있어요

baek님의 프로필 이미지
baek

작성한 질문수

질문하기