每日一题3：Pandas数据获取

一、每日一题

编写一个解决方案，选择 student_id = 101 的学生的 name 和 age 并输出。

解答：

import pandas as pd

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

题源：力扣 

二、总结 

        这里考察的是loc的用法，在我之前的博客中有写到。

.loc的用法：

介绍：

• loc是基于行标签（index）和列标签（columns）进行选择的。

• 语法：dataframe.loc[行标签, 列标签]/loc[index切片或列表，columns切片或列表]

• 可以同时选择多行和多列

逗号前面的是选择行标签，逗号后面的是选择列标签。