[LeetCode - Pandas] (Easy) 2880. Select Data

DataFrame students
+-------------+--------+
| Column Name | Type   |
+-------------+--------+
| student_id  | int    |
| name        | object |
| age         | int    |
+-------------+--------+​

student_id = 101인 학생의 이름과 나이를 선택하는 해결책을 작성하세요.

결과 형식은 아래 예시와 같습니다.

 

Example 1:
Input:
+------------+---------+-----+
| student_id | name    | age |
+------------+---------+-----+
| 101        | Ulysses | 13  |
| 53         | William | 10  |
| 128        | Henry   | 6   |
| 3          | Henry   | 11  |
+------------+---------+-----+
Output:
+---------+-----+
| name    | age | 
+---------+-----+
| Ulysses | 13  |
+---------+-----+

설명 : 학생 Ulysses는 student_id = 101을 가지고 있으며, 우리는 이름과 나이를 선택합니다.

 

✏️ 풀이

import pandas as pd

def selectData(students: pd.DataFrame) -> pd.DataFrame:
    return students[students['student_id'] == 101][['name', 'age']]