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