通过列名称子集数据框

Question

我的数据框的列包含“ pre_1，pre_2，pre_3等...”。 我也有标有“ post1，post2，post3 ...等”的发布数据。

有没有一种方法可以通过选择单词'pre'的所有列来在Python中切片数据？

谢谢。

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

---

要选择包含模式的列，请使用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