[판다스] 데이터프레임 다루기 : df.loc[], df.iloc[]

df = pd.DataFrame({'fruits':['사과','복숭아','바나나','오렌지','포도'],
                                 'animal':['강아지','병아리',np.nan,'팬더','고양이'], 'hobby':['독서','게임','달리기',np.nan,np.nan],
                                 'color':['green','red','purple','yellow','pink']})
df.columns = ['과일','동물','취미','색깔']
df.index = ['첫번째','두번','세번째','네번째','다섯번째']

입력 자료

1. loc으로 일부 행/열 추출

#---['과일','동물','취미'] 칼럼 추출
df1 = df.loc[:,['과일','동물','취미']] 

실행 결과

#---['첫번째','두번째','세번째','네번째'] index 추출
df2 = df1.loc[['첫번째','두번째','세번째','네번째'],:]
df2 = df1.loc[['첫번째','두번째','세번째','네번째'],]
df2 = df1.loc[['첫번째','두번째','세번째','네번째']]

실행 결과

#---['과일','동물','취미'] 칼럼, ['첫번째','두번째','세번째','네번째'] index 추출
df1 = df.loc[['첫번째','두번째','세번째','네번째'],['과일','동물','취미']]

실행 결과

2. iloc으로 일부 행/열 추출

#---[0,1,3] 번째 칼럼 추출
df1 = df.iloc[:,[0,1,3]]

실행 결과

#---[0,3,4] 번째 로우 추출
df1 = df.iloc[[0,3,4],:]
df1 = df.iloc[[0,3,4],]
df1 = df.iloc[[0,3,4],]

실행 결과

#---[0,1,3] 번째 칼럼, [0,3,4] 번째 로우 추출
df1 = df.iloc[[0,3,4],[0,1,3]]

 

실행 결과

 

3. 참고 : loc, iloc 이용한 Series와 DataFrame 출력 차이

df.loc[:,'과일']
df.iloc[:,0] #---Series로 출력

실행 결과

df.loc[:,['과일']]
df.iloc[:,[0]] #---DataFrame로 출력

실행 결과

df.loc['다섯번째',:]
df.loc['다섯번째',]
df.loc['다섯번째']
df.iloc[4,:]
df.iloc[4,]
df.iloc[4,] #---Series로 출력

 

실행 결과

df.loc[['다섯번째']]
df.iloc[[4]] #---DataFrame로 출력

실행 결과