[Python] 빠른 행/열 추출 take() 메소드

데이터프레임 df에서 행/열을 추출하는 방법은 다양하다.

 

1. iloc로 표현하는 방법

df.iloc[[0,10,20]]  # 행 추출
df.iloc(axis=1)[[0,10,20]]   # 열 추출

2. take로 표현하는 방법

df.take([0,10,20])   # 행 추출
df.take([0,10,20], axis=1)   # 열 추출

take를 통해서도 빠르게 추출할 수 있다.

하지만, take에서는 boolean index는 사용할 수 없고 그렇기 때문에 더 빠른 것 같다.

 

 

 

 

Reference :

- https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.take.html

pandas.DataFrame.take — pandas 1.3.4 documentation