[TIL] 데이터프레임에 모든 컬럼명 알고 싶을 때, 컬럼 split

데이터 프레임에 칼럼을 제거하거나 추가하고 싶을 때, 모든 칼럼명을 살펴보면 잘 활용할 수 있다. 이때

list(distance_result)

처럼 list(데이터프레임) 을 입력하면 모든 칼럼명들을 알 수 있다. 

 

---

 

distance_result라는 데이터 프레임이 있고, 그 안에 player_id라는 칼럼이 있다. player_id는 ABC_kwiseon 처럼 팀이름_닉네임의 형태를 가진다. 이때 팀만 잘라서 새로운 칼럼으로 만들고 싶다면 어떻게 해야할까? 

distance_result['team'] = distance_result.player_id.str.split('_').str[0]
print(distance_result)

df. 자르고 싶은 칼럼명. str.split('_'). str[0]으로 표현할 수 있다. str 형태를 split (=잘라주는데), '_' (=여기선 언더바를 기준으로!) 그리고. str[0]을 하면 0번째 인덱스 값, 즉 잘라낸 칼럼에서 맨 앞에를 가져오게 된다. 위의 예시에선 ABC만 가져온다.