Выбрать следующий столбец на основе ссылки на метку для столбцов кадра данных панд

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