595. Big Countries : leetcode

오유찬·2026년 1월 8일

pandas

목록 보기
3/9
import pandas as pd

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

loc 함수를 붙였을 때 오는 이점

  • 행, 열 동시 필터링 가능
  • 조회가 아닌 값 변경 시의 안전성
import pandas as pd

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

메모리 사용량 : 68.68MB
RUNTIME : 305ms

import pandas as pd

def big_countries(world: pd.DataFrame) -> pd.DataFrame:
    return world.loc[
        (world['area'] >= 3000000) | (world['population'] >= 25000000), # 행 선택
        ['name', 'population', 'area']

    ]

RUNTIME : 258ms
Memory : 69.00MB

profile
열심히 하면 재밌다

0개의 댓글