[python/파이썬] 데이터 전처리 - 레코드, 칼럼 추가 / 삭제

레코드, 칼럼 추가 / 삭제

import pandas as pd

df = pd.DataFrame({'a' : [1, 1, 3, 4, 5], 'b' : [2, 3, 2, 3, 4], 'c' : [3, 4, 7, 6, 4]})

df

1. 칼럼(column) 추가

• [] 연산자를 사용하여 기존 DataFrame에 새 열 추가 할 수 있다

-문제 : 1, 3, 6, 4. 8 로 이루어진 d 칼럼을 추가하기

df['d'] = [1, 3, 6, 4, 8]

df

-문제 : 1로 이루어진 e 칼럼 추가하기

df['e'] = [1, 1, 1, 1, 1]

df

이렇게 하는 방법도 있지만 조금 더 쉽게 하는 방법이 있다.

df['e'] = 1

df

df.dtypes

a  int64
b  int64
c  int64
d  int64
e  int64
dtype:  object

-문제 : a + b - c 의 결과로 이루어진 f 칼럼을 추가하기

df['f'] = df['a'] + df['b'] - df['c']

df

2. 칼럼 (column) 삭제

• drop(labels) : DataFrame 열 삭제

-문제 : 칼럼 d, e, f 를 삭제하기

df.drop(['d', 'e', 'f'], axis=1)

df

df.drop(['d', 'e', 'f'], axis=1, inplace=True)

df

3. 레코드 추가

• append() : 사전의 값을 행으로 직접 가지고 와서 DataFrame에 추가

-문제 : a에는 6, b에는 7, c에는 8을 추가하기

df.append({'a' : 6, 'b' : 7, 'c' : 8})

- ignore_index = True : 기존 index를 무시하고 싶을 때 , DataFrame을 합칠 때 ignore_index를 하지 않으면 데이터에 있던 인덱스를 그대로 가지고 옴

df.append({'a' : 6, 'b' : 7, 'c' : 8}, ignore_index = True)

df

df = df.append({'a' : 6, 'b' : 7, 'c' : 8}, ignore_index = True)

df

• loc[index] : 목록이 있는 데이터 프레임에 행을 추가

-문제 : a에는 7, b에는 8, c에는 9를 추가하기

df.loc[6] = [7, 8, 9]

df