從 pandas DataFrame 中刪除名稱包含特定字符串的第一個（或任何第 n 個）列

Question

假設我有一個 pandas DataFrame 如下：

df = pd.DataFrame([[1,2,3,4,5],[6,7,8,9,10]],columns=['a1693','b1124','b113','a2609', 'a1'])

例如，我想刪除b1124 。 我該怎么做？

我可以使用以下代碼將該列作為 pd.Series 獲取。

df.loc[:,df.columns.str.contains('b')].iloc[:,0]

但我不知道如何從df中刪除它。

另外，如果我想對多個列執行相同的操作，即刪除a1693和b1124 ，我該怎么做呢？

Answer 1

添加過濾器

df = df.drop(df.columns[df.columns.str.contains('b')][0],1)
   a1693  b113  a2609  a1
0      1     3      4   5
1      6     8      9  10

Answer 2

In [28]: N = 0

In [29]: df.drop(df.columns[np.where(df.columns.str.contains("b")==True)[0][N]], axis=1)
Out[29]:
   a1693  b113  a2609  a1
0      1     3      4   5
1      6     8      9  10

In [30]: N = 1

In [31]: df.drop(df.columns[np.where(df.columns.str.contains("b")==True)[0][N]], axis=1)
Out[31]:
   a1693  b1124  a2609  a1
0      1      2      4   5
1      6      7      9  10