pandas中关于DataFrame根据ID名称取出某列的数据集

除了可以用上一篇介绍的groupby方法以外,还可以使用df的query查询。
比如:

#查找UNIT_ID列中值为“10102”的所有行
res_df = df.query('UNIT_ID == "10102"')

query还可以级联使用:

#查找UNIT_ID列中值为“10102”并且GA_ID列中值为“910”的所有行
res_df = df.query('UNIT_ID == "10102"').query('GA_ID == "910"')

猜你喜欢

转载自blog.csdn.net/weekdawn/article/details/81391495