Python - Data Frame 값 추가/삭제

 지난 시간에는 Data Frame 내부의 값을 변경하는 방법을 정리하였습니다. 오늘은 Data Frame에 새로운 Column이나 Row를 추가 및 삭제해주는 방법을 정리하겠습니다.

 

 새로운 Row 또는 Column을 추가하는 방법은 아주 간단한데요, Row(Index)는 loc함수를 사용하여 새로운 Index를 지정하고 리스트를 사용해서 Column 수에 맞는 내부 값을 설정해주면 됩니다! Column을 추가하는 방법도 마찬가지로 새로운 Column을 지정하여 값을 설정해주면 추가가 됩니다. 이해하기 쉽도록 아래 예를 살펴보겠습니다.

 

Galaxy 10E Index를 추가

 

제조사 Column을 추가

 

 이제 반대로 기존의 Data Frame에 존재하던 Row 또는 Column을 삭제하는 방법을 살펴보겠습니다. drop함수를 사용해서 삭제가 가능한데요, DataFrame.drop(Index, axis="index", inplace=True or False)로 삭제할 수 있습니다.

drop함수 안을 하나씩 살펴보겠습니다.

 

 먼저 제거하고자 하는 Index 또는 Row를 처음 파라미터로 넣고, axis에는 제거하는 것이 Row(index)라면 index, Column이라면 columns를 적어 줍니다. inplace에는 원본 Data Frame을 바꾸려면 True, 바꾸지 않으려면 False를 넣어주면 됩니다. 아래 예를 살펴보겠습니다.

 

Galaxy 10E Index 삭제

 

 위의 예에서는 Galaxy 10E라는 Index를 삭제하는데요, 첫 파라미터는 Galaxy 10E로 넣고, axis는 index를 넣어 줍니다. 5번 Cell에서는 inplace에 False를 넣었고 df를 출력할 경우 Galaxy 10E가 그대로 있습니다. 6번 Cell과 함께 inplace에 Ture를 넣을 경우 df를 출력시 Galaxy 10E가 제거된 것을 확인할 수 있습니다.

 

 한번에 여러 개의 index 및 Column을 삭제할 수도 있습니다. 첫 번째 파라미터로 하나의 Row/Column이 아닌 리스트를 넣어주어 여러 index 및 Column을 삭제할 수 있습니다.

 

 

 이렇게 Data Frame에 Column과 Row를 추가 및 삭제하는 것을 정리해보았습니다. 이번 내용을 정리하면서도 느낀 점은 역시 Data Frame의 인덱싱이 중요하다는 것을 다시 한번 느낄 수 있었습니다.