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

题目描述

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

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

测试用例

输入：

输出

解析

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

import pandas as pd

def big_countries(world: pd.DataFrame) -> pd.DataFrame:
    df:pd.DataFrame = world[(world["area"]>=3000000)|(world["population"]>=25000000)]
    return df[["name", "population", "area"]]

注意在提取pd.Dataframe时需使用双方括号

也可以使用pandas自带的loc函数进行简化

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

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