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

文章pandas笔记（一）-- 大的国家（逻辑索引、切片）分享给大家，欢迎收藏Python资料网，专注分享技术知识

题目描述

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

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

测试用例

输入：

name	continent	area	population	gdp
Afghanistan	Asia	652230	25500100	20343000000
Albania	Europe	28748	2831741	12960000000
Algeria	Africa	2381741	37100000	188681000000
Andorra	Europe	468	78115	3712000000
Angola	Africa	1246700	20609294	100990000000

输出

name	population	area
Afghanistan	25500100	652230
Algeria	37100000	2381741

解析

本题考查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']]

---

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

版权声明：本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若内容造成侵权/违法违规/事实不符，请联系邮箱：jacktools123@163.com进行投诉反馈，一经查实，立即删除！