[LeetCode - Pandas] (Easy) 2891. Method Chaining

DataFrame animals
+-------------+--------+
| Column Name | Type   |
+-------------+--------+
| name        | object |
| species     | object |
| age         | int    |
| weight      | int    |
+-------------+--------+

 

이 문제는 "100킬로그램보다 더 무거운 동물들의 이름을 리스트로 반환하고, 그 동물들을 체중을 기준으로 내림차순으로 정렬"하는 문제입니다.

 

예시를 들어 설명하자면, 주어진 동물들의 체중 데이터에서 100킬로그램을 초과하는 동물들의 이름을 출력해야 하고, 그 동물들을 체중이 무겁게 내림차순으로 정렬해야 합니다.

 

정답

import pandas as pd

def findHeavyAnimals(animals: pd.DataFrame) -> pd.DataFrame:
    df = animals[animals['weight'] > 100]
    result = df.sort_values(by='weight', ascending=False)

    return result[['name']]