如何通过regex从dataframe中选择列?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (172)

我在python pandas里的dataframe如下:

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

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

提问于
用户回答回答于

使用:

import pandas as pd

df = pd.DataFrame([[10, 14, 12, 44, 45, 78]], columns=['a', 'b', 'c', 'd1', 'd2', 'd3'])

df.select(lambda col: col.startswith('d'), axis=1)

结果:

   d1  d2  d3
0  44  45  78
用户回答回答于
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

扫码关注云+社区

领取腾讯云代金券