[LeetCode - Pandas] (Easy) 2886. Change Data Type

DataFrame students
+-------------+--------+
| Column Name | Type   |
+-------------+--------+
| student_id  | int    |
| name        | object |
| age         | int    |
| grade       | float  |
+-------------+--------+

오류를 수정하는 해결 방법을 작성하세요:

'grade' 열이 실수(float)로 저장되어 있으므로 이를 정수(int)로 변환하세요.

결과 형식은 아래 예시와 같습니다.

 

Example 1:
Input:
DataFrame students:
+------------+------+-----+-------+
| student_id | name | age | grade |
+------------+------+-----+-------+
| 1          | Ava  | 6   | 73.0  |
| 2          | Kate | 15  | 87.0  |
+------------+------+-----+-------+
Output:
+------------+------+-----+-------+
| student_id | name | age | grade |
+------------+------+-----+-------+
| 1          | Ava  | 6   | 73    |
| 2          | Kate | 15  | 87    |
+------------+------+-----+-------+

설명:
'grade' 열의 데이터 타입이 int로 변환되었습니다.

 

풀이

import pandas as pd

def changeDatatype(students: pd.DataFrame) -> pd.DataFrame:
    students['grade'] = students['grade'].astype(int)

    return students

 

개념 정리

astype : 열의 요소의 데이터 타입을 변경하는 함수

df.astype(dtype, copy=True, errors='raies')

dtype : 변경할 데이터 타입

copy : 사본을 생성할지 여부

errors : raies / ignore 변경 불가 시 오류를 발생시킬지 여부