[LeetCode - Pandas] (Easy) 2885. Rename Columns

DataFrame students
+-------------+--------+
| Column Name | Type   |
+-------------+--------+
| id          | int    |
| first       | object |
| last        | object |
| age         | int    |
+-------------+--------+

다음과 같이 열 이름을 변경하는 해결책을 작성해 주세요:

- id → student_id

- first → first_name

- last → last_name

- age → age_in_years

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

Example 1:
Input:
+----+---------+----------+-----+
| id | first   | last     | age |
+----+---------+----------+-----+
| 1  | Mason   | King     | 6   |
| 2  | Ava     | Wright   | 7   |
| 3  | Taylor  | Hall     | 16  |
| 4  | Georgia | Thompson | 18  |
| 5  | Thomas  | Moore    | 10  |
+----+---------+----------+-----+
Output:
+------------+------------+-----------+--------------+
| student_id | first_name | last_name | age_in_years |
+------------+------------+-----------+--------------+
| 1          | Mason      | King      | 6            |
| 2          | Ava        | Wright    | 7            |
| 3          | Taylor     | Hall      | 16           |
| 4          | Georgia    | Thompson  | 18           |
| 5          | Thomas     | Moore     | 10           |
+------------+------------+-----------+--------------+

 

설명: 열 이름이 그에 맞게 변경됩니다.

 

✏️ 풀이

import pandas as pd

def renameColumns(students: pd.DataFrame) -> pd.DataFrame:
    students.rename(columns={'id':'student_id', 'first':'first_name', 'last':'last_name', 'age':'age_in_years'}, inplace=True)
    
    return students

 

📚 개념 정리

rename() ? 컬럼, 인덱스 이름 변경

df.rename(columns={'원래컬럼명':'수정컬럼명', ...}, inplace=True)

df.rename(index={0:'수정인덱스명', ...})

inplace : 원본을 바로 수정할지 여부