import pandas as pd
def selectData(students: pd.DataFrame) -> pd.DataFrame:
100015. Select Data
DataFrame students +-------------+--------+ | Column Name | Type | +-------------+--------+ | student_id | int | | name | object | | age | int | +-------------+--------+
Write a solution to select the name and age of the student with student_id = 101
.
The result format is in the following example.
Example 1: Input: +------------+---------+-----+ | student_id | name | age | +------------+---------+-----+ | 101 | Ulysses | 13 | | 53 | William | 10 | | 128 | Henry | 6 | | 3 | Henry | 11 | +------------+---------+-----+ Output: +---------+-----+ | name | age | +---------+-----+ | Ulysses | 13 | +---------+-----+ Explanation: Student Ulysses has student_id = 101, we select the name and age.
原站题解
pythondata 解法, 执行用时: 280 ms, 内存消耗: 59.6 MB, 提交时间: 2023-10-07 10:32:08
''' 简单过滤 ''' import pandas as pd def selectData(students: pd.DataFrame) -> pd.DataFrame: return students[students['student_id'] == 101][['name', 'age']]