[python/파이썬] 데이터 전처리 - 데이터 프레임 생성, 칼럼명 추출/변경

데이터 전처리

1) 데이터 프레임 생성

1. dict 를 이용한 데이터 프레임 생성

import pandas as pd

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

type(df)

pandas.core.frame.DataFrame

df

dummy = {'a': [1, 2, 3], 'b' : [4, 5, 6], 'c' : [7, 8, 9]}

df2 = pd.DataFrame(dummy)

df2

2. List 를 이용한 데이터 프레임 생성

a = [[1, 4, 7], [2, 5, 8], [3, 6, 9]]

df3 = pd.DataFrame(a)

df3

df3.columns = ['a', 'b', 'c']

df3

• 문제 : 아래 테이블과 같은 데이터 프레임 만들기

a = {'company' : ['abc', '회사', 123], '직원수' : [400, 10, 6]}

df4 = pd.DataFrame(a)

df4

• 문제 : 아래 테이블과 같은 데이터 프레임 만들기

a = {'company' : ['abc', '회사', 123], '직원수' : [400, 10, 6], '위치' : ['Seoul', NaN, 'Busan']}

a = {'company' : ['abc', '회사', 123], '직원수' : [400, 10, 6], '위치' : ['Seoul', , 'Busan']}

-numpy 를 통한 해결

import numpy as np

a = {'company' : ['abc', '회사', 123], '직원수' : [400, 10, 6], '위치' : ['Seoul', np.NaN, 'Busan']}

df5 = pd.DataFrame(a)

df5

2) 칼럼명 추출/ 변경

1. 데이터 프레임 생성

import pandas as pd

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

df

2. 칼럼명 얻기

df.columns

Index(['a','b','c'], dtype='object')

df.colums[1]

'b'

• 문제 : 칼럼명인 a, b, c를 d, e, f로 바꾸어라

1. 치환을 통한 칼럼명 변경

df.columns = ['d', 'e', 'f']

df

• 문제 : 칼럼명인 d, e, f 중 d를 '디' 로 f를 '에프'로 바꾸어라

df.colums = ['디', 'e', '에프']

df

2. rename을 통한 칼럼명 변경

# 데이터 프레임 재생성
df = pd.DataFrame({'a': [1, 2, 3], 'b' : [4, 5, 6], 'c' : [7, 8, 9]})
df.columns = ['d', 'e', 'f']

df

df.rename(columns = {'d' : '디', 'f' : '에프'})

df

rename 을 통해서 컬럼명이 변경되었지만, 저장은 안 됨.

-inplace = True 로 되어 있어야 저장됨

df.rename(columns = {'d' : '디', 'f' : '에프'}, inplace = True)

df