python - 如何通过正则表达式从数据框中选择列

我在python Pandas 中有一个数据框。数据框的结构如下:

   a    b    c    d1   d2   d3
   10   14   12   44  45    78

我想选择以d开头的列。有没有一种简单的方法可以在python中实现这一点。

最佳答案

您可以通过以下方式使用 DataFrame.filter :

import pandas as pd

df = pd.DataFrame(np.array([[2,4,4],[4,3,3],[5,9,1]]),columns=['d','t','didi'])
>>
   d  t  didi
0  2  4     4
1  4  3     3
2  5  9     1

df.filter(regex=("d.*"))

>>
   d  didi
0  2     4
1  4     3
2  5     1

这个想法是通过regex选择列