Выберите столбцы по индексу, имя которого содержит подстроку

Question

У меня есть кадр данных со столбцами, похожими на

Run  Date1  Date2  Date3  A.0  A.1  A.2  B.0  B.1  B.2....

Я хочу сохранить столбцы Run и Date, но выбрать только столбцы, которые содержат A, B и т. Д. Таким образом, вывод должен выглядеть примерно так

Run  Date1  Date2  Date3  A.0  A.1  A.2 

Использование фильтра (например, = A, axis = 1) дает мне только столбцы с A, а не столбцы прогона и даты

Я пытался

df.iloc[:, [0,1,2,3,df.columns.get_loc(df.columns.str.contains("A"))]]

, ноон говорит, что логический список не является допустимым ключом

Answer 1

попробуйте

cols = ["Run", "Date1", "Date2", "Date3"] + [col for col in df.columns if col.startswith["A"] or col.startswith["B"]]

df.loc[:, cols]