[파이썬]Pandas 행과 열 선택하기

행, 열 선택/필터 하기¶

In [7]:

import pandas as pd
people = [
    {'name': 'Kim', 'age': 15, 'job':'student'},
    {'name': 'Lee', 'age': 34, 'job':'doctor'},
    {'name': 'Choi', 'age': 21, 'job':'warrior'}
]
df = pd.DataFrame(people)

In [8]:

df

Out[8]:

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

먼저 데이터프레임을 저장해주었습니다.

행 선택하기¶

In [9]:

df[1:3] # 범위 : 1이상 3이하

Out[9]:

	name	age	job
1	Lee	34	doctor
2	Choi	21	warrior

In [16]:

df.loc[ [0,2] ] # 범위가 아니라 특정 행을 선택하고 싶을 때

Out[16]:

	name	age	job
0	Kim	15	student
2	Choi	21	warrior

열 선택하기¶

In [17]:

df['name']

Out[17]:

0     Kim
1     Lee
2    Choi
Name: name, dtype: object

위의 경우에는 시리즈타입으로 반환되는 것을 주의해주세요!

In [18]:

df[ ['name','age']]

Out[18]:

	name	age
0	Kim	15
1	Lee	34
2	Choi	21

In [23]:

df.loc[:, ['name','age']]

Out[23]:

	name	age
0	Kim	15
1	Lee	34
2	Choi	21

In [26]:

df.iloc[:, :2]

Out[26]:

	name	age
0	Kim	15
1	Lee	34
2	Choi	21

조건에 따라 선택하기¶

In [11]:

df[df.age > 25]

Out[11]:

	name	age	job
1	Lee	34	doctor

In [13]:

df[ (df.age < 25) & (df.name == 'Kim') ]

Out[13]:

	name	age	job
0	Kim	15	student

In [27]:

df.filter(items=['age', 'job'])

Out[27]:

	age	job
0	15	student
1	34	doctor
2	21	warrior

열이름에 a가 들어가는 열만 선택하고 싶을 때

In [28]:

df.filter(like='a', axis=1)

Out[28]:

	name	age
0	Kim	15
1	Lee	34
2	Choi	21

열이름이 b로 끝나는 열만 선택하고 싶을 때

In [29]:

df.filter(regex='b$', axis=1)

Out[29]:

	job
0	student
1	doctor
2	warrior