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

Question

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

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

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

Answer 1

您可以使用 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

选择列