열 이름으로 데이터 프레임 부분 집합

Question

"pre_1, pre_2, pre_3 등 ..."이 포함 된 열이있는 데이터 프레임이 있습니다. 또한 포스트 데이터 post1, post2, 포스트 3 ... 등) "으로 표시했다.

단어 '사전'의 모든 열을 선택하여 파이썬에서 데이터를 슬라이스 할 수있는 방법이 있습니까?

감사합니다.

Answer 1

당신은 열 이름의 패턴과 일치하는 like 또는 regex 키워드로 filter를 사용할 수 있습니다

df = pd.DataFrame({ 
    'pre_1': [1,2], 
    'pre_2': [3,4], 
    'pre_3': [5,6], 
    'post1': [7,8], 
    'post2': [9,10], 
    'post3': [11,12] 
}) 

df 
#post1 post2 post3 pre_1 pre_2 pre_3 
#0 7  9  11  1  3  5 
#1 8  10  12  2  4  6 

---

이 번째 열을 선택하려면 다음은 pre로 그 (^)을 시작 열을 선택합니다

df.filter(like='pre') 

#pre_1 pre_2 pre_3 
#0 1  3  5 
#1 2  4  6 

---

또는 당신은 정규 표현식을 사용할 수 있습니다 : 패턴을 포함에서의 like 키워드를 사용

df.filter(regex='^pre') 

#pre_1 pre_2 pre_3 
#0 1  3  5 
#1 2  4  6