按列名称设置数据框

Question

我有一个包含“pre_1，pre_2，pre_3等...”的列的数据框。我也有后标记数据为“POST1，POST2，post3 ...等）。

有没有用这个词‘前’选择所有列切片在Python数据？

感谢。

Answer 1

您可以使用filter与like或regex关键字列名匹配的模式：

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 

---

要选择个列在包含一个模式，使用like关键字：

df.filter(like='pre') 

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

---

或者你可以使用正则表达式，下面将pre选择列启动（^）：

df.filter(regex='^pre') 

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