一、每日一题
编写一个解决方案,选择 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切片或列表]
-
可以同时选择多行和多列
逗号前面的是选择行标签,逗号后面的是选择列标签。