python - 根据对 pandas 数据框列的标签引用选择下一列

Question

import pandas as pd, numpy as np

df1 = pd.DataFrame(np.random.randint(0,100,size=(10, 4)), columns=['a','qww12','b','hdbh'])

如果我必须选择列“a”和“b”，我会执行以下操作:

sublist=['a','b']

df2= df1[df1.columns.intersection(sublist)]

有没有一种更简单的方法可以选择“a”和“b”旁边的列，而无需键入名称？实际上，我正在从 Excel 工作表中获取数据，名称可以更改并且可以有空格。

提前致谢。

预期输出

   qww12  hdbh
0     18    86
1     84    67
2     23    45
3     60    15
4     82    60
5     84    75
6     24    23
7     34    79
8      4    45
9      2    82

Answer 1

您可以使用get_indexer()为此，要获取 sublist 的索引并添加 1 以获取下一列索引，然后使用 df.iloc[] :

df1.iloc[:,df1.columns.get_indexer(sublist)+1]

<小时/>

   qww12  hdbh
0     48    91
1     86    67
2     21    98
3     11    94
4      9    14
5     70    54
6     38    61
7     99    65
8     16    86
9     15    40