Pandas

Vista general

Este problema nos pide retornar las primeras 3 filas del DataFrame employees.

Conceptos clave:

1. DataFrame: una estructura en forma de una tabla de 2 dimensiones, similar a una hoja de cálculo o una tabla SQL. Cada fila representa un registro individual y cada columna un atributo diferente. Se puede redimensionar y está diseñada para manejar una mezcla de diferentes tipos de información.
2. Método head: un médodo proporcionado por pandas que es usado para retornar las primeras n filas de un DataFrame. Si se omite n, retorna las primeras 5 filas. Este método es util para obtener una pequeña muestra o echar un vistazo rápido de la información almacenada al principio de DataFrames gigantes.

Intuición

Veamos, paso a paso, cómo retornar las primeras 3 filas del DataFrame.

1. Importar pandas:

import pandas as pd

Esta línea importa la libreria pandas y le asigna un alias pd. La libreria pandas proporciona estructuras de datos rápidas, flexibles y expresivas diseñadas para trabajar con datos estructurados (tabulares, multidimensionales, potencialmente heterogéneos).

2. Utilizar head:

Veamos un ejemplo para ver cómo podemos usar head para resolver este problema.

Dado el siguiente DataFrame:

employee_id	name	department	salary
3	Bob	Operations	48675
90	Alice	Sales	11096
9	Tatiana	Engineering	33805
60	Annabelle	InformationTechnology	37678
49	Jonathan	HumanResources	23793
43	Khaled	Administration	40454

Retornamos el DataFrame usando la función head con el número 3 como argumento, para indicar que nos interesa filtrar las primeras 3 filas:

return employees.head(3)

El DataFrame resultante es:

employee_id	name	department	salary
3	Bob	Operations	48675
90	Alice	Sales	11096
9	Tatiana	Engineering	33805

Visualización de la función head aplicada al DataFrame employees:

Implementación

import pandas as pd


def selectFirstRows(employees: pd.DataFrame) -> pd.DataFrame:
    return employees.head(3)