Display the First Three Rows

DataFrame: employees
+-------------+--------+
| Column Name | Type   |
+-------------+--------+
| employee_id | int    |
| name        | object |
| department  | object |
| salary      | int    |
+-------------+--------+

Write a solution to display the first 3 rows of this DataFrame.

Example 1:

Input:
DataFrame employees
+-------------+-----------+-----------------------+--------+
| 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  |
+-------------+-----------+-----------------------+--------+
Output:
+-------------+---------+-------------+--------+
| employee_id | name    | department  | salary |
+-------------+---------+-------------+--------+
| 3           | Bob     | Operations  | 48675  |
| 90          | Alice   | Sales       | 11096  |
| 9           | Tatiana | Engineering | 33805  |
+-------------+---------+-------------+--------+
Explanation: 
Only the first 3 rows are displayed.

Solution

import pandas as pd

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


if __name__ == '__main__':
    data = [[3, 'Bob', 'Operations', 48675],
            [90, 'Alice', 'Sales', 11096],
            [9, 'Tatiana', 'Engineering', 33805],
            [60, 'Annabelle', 'InformationTechnology', 37678],
            [49, 'Jonathan', 'HumanResources', 23793],
            [43, 'Khaled', 'Administration', 40454],
            ]
    employee = pd.DataFrame(data, columns=['employee_id', 'name', 'department', 'salary'])
    print(selectFirstRows(employee))

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!