해결된 질문
작성
·
207
0
# object명의 컬럼을 아예 없애버리기
cols = train.select_dtypes(exclude = 'object').columns
cols
train = train[cols]
train.head()
답변 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'를 활용하는 방법이 있어요
list()는 묶어도 되고 안묶어도 됩니다.