본문 바로가기

🐍Python

[TIL] 변수에 값 할당하기 assign 함수

지난 시간에 살짝 맛본 python assign 함수 복습하기 

assign 함수는 복합 대입 연산자로, 연산과 할당을 합쳐놓은 함수다. 변수가 이전에 가졌던 값을 수정해 할당하는 일에 쓸 수 있으며, 너무 많이 쓰다 보면 코드를 읽기가 어려워질 수도 있으니 사용에 주의해야 한다.

distance_1 = player_summary.assign(total_distance = player_summary['ride_distance'] + player_summary['walk_distance'])
distance_1.head()

 

위 코드를 내 방식대로 해석하면 
새로운 데이터 = 기존 데이터.assign(새로운 변수명 = 기존 데이터['수정/할당하려는 칼럼명'] +-*/(연산)로 만들 수 있다.
변수끼리 더하거나, 나누는 등 연산을 포함할 수 있는 것이다. 

distance_result = distance_1.assign(dis_car = distance_1['ride_distance'] / distance_1['total_distance'] *100,
                                    dis_walk = distance_1['walk_distance'] / distance_1['total_distance']*100).sort_values('team_rank')

 

sort_values와 함께 사용하면 기준을 잡고 데이터를 뽑을 수 있어 활용도가 높다.