[Python/Pandas] 리트코드 1667. Fix Names in a Table

문제

Table: Users
+----------------+---------+
| Column Name    | Type    |
+----------------+---------+
| user_id        | int     |
| name           | varchar |
+----------------+---------+
user_id is the primary key (column with unique values) for this table.
This table contains the ID and the name of the user. The name consists of only lowercase and uppercase characters.

Write a solution to fix the names so that only the first character is uppercase and the rest are lowercase.

Return the result table ordered by user_id.

 

https://leetcode.com/problems/fix-names-in-a-table/description/?lang=pythondata

예시

Input: 
Users table:
+---------+-------+
| user_id | name  |
+---------+-------+
| 1       | aLice |
| 2       | bOB   |
+---------+-------+
Output: 
+---------+-------+
| user_id | name  |
+---------+-------+
| 1       | Alice |
| 2       | Bob   |
+---------+-------+

문제 풀이

import pandas as pd

def fix_names(users: pd.DataFrame) -> pd.DataFrame:
    # name 열 capitalize
    users['name'] = users['name'].str.capitalize()
    
    # user_id 기준 정렬
    result = users.sort_values('user_id')
    return result

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.