목록2025/01/19 (1)
susinlee 님의 블로그
111. Last Person to Fit in the Bus
[문제]https://leetcode.com/problems/last-person-to-fit-in-the-bus/description/ [풀이]1. turn을 기준으로 정렬을 해준 후에2. weight 열의 누적합을 구해서 새로운 컬럼을 생성한다.3. 누적합이 1000이하인 행들만 필터링한 뒤 가장 마지막행을 가져와서 반환하면 Pandasimport pandas as pddef last_passenger(queue: pd.DataFrame) -> pd.DataFrame: queue.sort_values('turn', inplace=True) queue['cum_sum'] = queue.weight.cumsum() return queue[queue.cum_sum SQLSELECT ..
코드카타/Pandas
2025. 1. 19. 11:13