■ select 함수를 사용해 특정 조건에 해당하는 값으로 신규 컬럼을 추가하는 방법을 보여준다.
▶ main.py
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import pandas as pd import numpy as np dataFrame = pd.DataFrame( { "name" : ["김철수", "이영희", "박민수", "정지은"], "age" : [25, 30, 35, 28], "city" : ["서울", "부산", "서울", "대전"], "score" : [85, 92, 78, 95] } ) print(dataFrame) """ name age city score 0 김철수 25 서울 85 1 이영희 30 부산 92 2 박민수 35 서울 78 3 정지은 28 대전 95 """ print() conditionList = [ (dataFrame["age" ] < 30) & (dataFrame["score"] >= 90), (dataFrame["age" ] >= 30) & (dataFrame["score"] >= 90), (dataFrame["score"] < 90) ] choiceList = ["A+", "A", "B"] dataFrame["grade"] = np.select(conditionList, choiceList, default = "C") print(dataFrame) """ name age city score grade 0 김철수 25 서울 85 B 1 이영희 30 부산 92 A 2 박민수 35 서울 78 B 3 정지은 28 대전 95 A+ """ |
▶ requirements.txt
|
1 2 3 4 5 6 7 8 |
numpy==2.1.2 pandas==2.2.3 python-dateutil==2.9.0.post0 pytz==2024.2 six==1.16.0 tzdata==2024.2 |
※ pip install pandas 명령을 실행했다.