每日一题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切片或列表]

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

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