2879_display_the_first_three_rows.py

# Problem 2879: Display the First Three Rows
# https://leetcode.com/problems/display-the-first-three-rows/

import pandas as pd


def select_first_rows(employees: pd.DataFrame) -> pd.DataFrame:
    # 使用 head 決定顯示前幾 row
    return employees.head(3)