Pandas DataFrame 11-03. object 열의 적절 dtype추론(infer_objects)

11-03. object 열의 적절 dtype추론(infer_objects)

DataFrame.infer_objects()

 

개요

---

infer_object메서드는 dtype이 object인 열에 대해서 적당한 dtype을 추론합니다.
사용법을 참고 바랍니다.

 

사용법

---

기본 사용법
df.infer_object( )

반응형

 

예시

---

먼저 str과 int가 혼합된 col1을 가진 DataFrame 객체를 만들어 dtype이 object인 열을 만들어 보겠습니다.

col1 = ['a','b', 3, 4]
df = pd.DataFrame({'col1':col1},index=['row1','row2','row3','row4'])
print(df)
>>
     col1
row1    a
row2    b
row3    3
row4    4

print(df.dtypes)
>>
col1    object
dtype: object

 

---

이제 df에서 형식이 int인 행만 남겨서 인덱싱을 한 뒤. dtype을 살펴보면, 여전히 dtype이 object인 것을 확인 할 수 있습니다.

df = df.iloc[2:]
print(df)
>>
     col1
row3    3
row4    4

print(df.dtypes)
>>
col1    object
dtype: object

이런 경우에 대해 infer_object는 가장 적당한 dtype을 제안하는 기능을 합니다.

print(df.infer_objects())
>>
      col1
row3     3
row4     4

print(df.infer_objects().dtypes)
>>
col1    int64
dtype: object

위와 같이 int형식만 남은 df의 col1 열에 대해 가장 적절한 dtype인 int64로 변환된 것을 확인할 수 있습니다.