[파이썬]Pandas DataFrame만들기

데이터프레임만들기¶

In [1]:

import pandas as pd

먼저 개인적으로 가장 직관적이라 생각되는 방법부터 알려드리겠습니다.
데이터로 사용할 내용을 리스트와 딕셔너리를 이용해 만들어줍니다.
여기서 키는 열의 이름을 나타냅니다.

In [2]:

people = [
    {'name': 'Kim', 'age': '15', 'job':'student'},
    {'name': 'Lee', 'age': '34', 'job':'doctor'},
    {'name': 'Choi', 'age': '21', 'job':'warrior'}
]

In [3]:

df = pd.DataFrame(people)

In [4]:

df.head()

Out[4]:

	name	age	job
0	Kim	15	student
1	Lee	34	doctor
2	Choi	21	warrior

만약 열의 순서를 바꾸고 싶다면?
아래와 같이 열의 순서를 바꿔서 다시 저장해주면 됩니다. 괄호가 두 개입니다.

In [5]:

df = df[['age', 'job', 'name']]

In [6]:

df

Out[6]:

	age	job	name
0	15	student	Kim
1	34	doctor	Lee
2	21	warrior	Choi

---

이번에는 리스트만을 이용해서 만들어보겠습니다.

In [9]:

people_list = [
    ['Kim', 15, 'student'],
    ['Lee', 34, 'doctor'],
    ['Choi', 21, 'warrior']
]

In [10]:

column_list = ['name', 'age', 'job']

In [11]:

df = pd.DataFrame.from_records(people_list, columns = column_list)

In [12]:

df

Out[12]:

	name	age	job
0	Kim	15	student
1	Lee	34	doctor
2	Choi	21	warrior