pandas条记（一）-- 大的国家（逻辑索引、切片）

题目描述

如果一个国家满足下述两个条件之一，则认为该国是 大国 ：

• 面积至少为 300 万平方公里
  
• 人口至少为 2500 万
  编写办理方案找出大国的国家名称、人口和面积
  按任意顺序返回结果表，如下例所示
  

测试用例

输入：
namecontinentareapopulationgdpAfghanistanAsia6522302550010020343000000AlbaniaEurope28748283174112960000000AlgeriaAfrica238174137100000188681000000AndorraEurope468781153712000000AngolaAfrica124670020609294100990000000输出
namepopulationareaAfghanistan25500100652230Algeria371000002381741解析

本题考察pandas的索引操作，属于基础题（最基础的那种），根本思绪就是先筛选，后切片

1. import pandas as pd
3. def big_countries(world: pd.DataFrame) -> pd.DataFrame:
4.     df:pd.DataFrame = world[(world["area"]>=3000000)|(world["population"]>=25000000)]
5.     return df[["name", "population", "area"]]

复制代码

留意在提取pd.Dataframe时需使用双方括号
也可以使用pandas自带的loc函数进行简化

1. import pandas as pd
2. def big_countries(world: pd.DataFrame) -> pd.DataFrame:
3.   return world.loc[(world['area'] >= 3000000) | (world['population'] >= 25000000), ['name', 'population', 'area']]

复制代码

小白一枚，于本日正式开通个人博客，特此纪念，在此祝愿园子越办越好

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。