Corrige los nombres en una tabla

Escribe una solución para corregir los nombre de tal forma que sólo la primera letra sea mayúscula y el resto minúsculas. Regresa el resultado ordenado por `user_id`.

#pandas

Tabla: Users

+----------------+---------+
| Column Name    | Type    |
+----------------+---------+
| user_id        | int     |
| name           | varchar |
+----------------+---------+
- `user_id` es la llave primaria (columna con valores únicos) para esta tabla.
- Esta tabla contiene el ID y el nombre de cada usuario. El nombre solo contine letras minúsculas y mayúsculas.

El formato del resultado se muestra en el siguiente ejemplo.

Ejemplo 1:

Entrada:

Tabla Users:
+---------+-------+
| user_id | name  |
+---------+-------+
| 1       | aLice |
| 2       | bOB   |
+---------+-------+

Salida:

+---------+-------+
| user_id | name  |
+---------+-------+
| 1       | Alice |
| 2       | Bob   |
+---------+-------+

Solución

import pandas as pd

def fix_names(users: pd.DataFrame) -> pd.DataFrame:
    users = users.sort_values(by='user_id')
    users.name = users.name.str.capitalize()
    return users

slackmart blog © 2024