7. 컬럼을 추가하고 삭제하기(drop)

이번에는 컬럼을 추가하고 삭제해보겠습니다. 

엑셀에서는 단지 우클릭하고 열을 삽입해주면 되었습니다.

엑셀에서 열 삽입

python에서는 열의 이름을 정하고, 값을 넣어주면 열이 추가됩니다. 

간단하게 tmp 열을 만들고 1 값을 넣어주었습니다.

boston_price['tmp']=1

테스트 삼아 넣어본 열이니, 다시 삭제하겠습니다. 엑셀은 삽입할 때와 마찬가지로 삭제 위치에 우클릭하고 열을 삭제하면 됩니다.

엑셀에서 열 삭제

python에서는 drop 함수를 사용합니다.

boston_price.drop('tmp',axis=1,inplace=True)

DataFrame에 drop() 함수를 붙이고, 인자로 삭제할 컬럼명('tmp'), 삭제 방향(axis=1), DataFrame에 즉시 반영 여부(inplace=True)를 입력하면 됩니다.

이때, axis=1이 열 방향을 의미하고, 행 방향은 axis=0으로 씁니다.

inpalce=True 옵션을 주면 DataFrame에 즉시 반영됩니다.

아래와 같이 boston_price에 할당해주어도 동일한 결과를 얻을 수 있습니다.

boston_price=boston_price.drop('tmp',axis=1)

만약 여러 개 열을 삭제한다면, 컬럼명을 리스트로 넣어주면 됩니다.

boston_price.drop(['tmp','tmp2'],axis=1,inplace=True)

이상으로 Python에서 컬럼 추가와 삭제에 대해 알아보았습니다.