Python/프로그래머스

[LV1] 데이터 분석 - PCCE 기출문제

헬로희 2024. 11. 28. 22:17
728x90

* 저 만의 문제풀이라 깔끔하지 않을 수도 있으니 참고부탁드립니다.

Q. 코드번호(code), 제조일(date), 최대 수량(maximum), 현재 수량(remain)으로 구성된 데이터(data)들 중
기준값(val_ext)과 기준이 되는 열(ext)에 해당하는 행을 뽑아 정렬기준(sort_by)으로 오름차순 정렬하기

  • ext와 sort_by는 "code", "date", "maximum", "remain" 중 하나
def solution(data, ext, val_ext, sort_by):
    import pandas as pd
    df_data=pd.DataFrame(data, columns=["code","date","maximum","remain"])
    result=df_data[df_data[ext]<=val_ext].sort_values(by=sort_by, ascending=True)   
    answer = result.values.tolist()
    return answer

 

728x90