Python / Pandas - 刪除以字符串開頭的列

Question

我有這樣的數據幀

id    2013 Profits   2001 Revenues    1999 Assets...
31    xxxx           xxxx             xxxx
...

我想刪除不以'201'開頭的列（我只想保留數據2010和轉發）。

我怎樣才能做到這一點？

Answer 1

使用pd.DataFrame.filter

df.filter(like='201')

   2013 Profits
id             
31         xxxx

---

正如@StevenLaan指出的那樣，使用like會包含一些列，這些列在列名中的其他位置具有模式字符串。 我們可以確保只使用regex獲取以模式字符串開頭的列。

df.filter(regex='^201')

   2013 Profits
id             
31         xxxx

Answer 2

這將確保僅保留以201開頭的列。

df[[c for c in df.columns if c.startswith('201')]]

同樣可以實現

df.filter(regex='^201')

Answer 3

另一種方式：

In [16]: df.loc[:, df.columns.str.contains('^201')]
Out[16]:
  2013 Profits
0         xxxx