Отфильтровать столбцы, содержащие определенное слово

Question

В pandas DataFrame Мне нужно отфильтровать те столбцы, которые содержат слово «fixed». Затем я хочу создать еще один DataFrame только с этими столбцами. Как я могу это сделать?

cols = dataset.columns 

split = lambda x: [c for c in x.split('_') if c.startswith('fixed')] 

Answer 1

Просто используйте .ix атрибут:

fixed = [c for c in dataset.columns if c.startswith("fixed")] 
fixed_dataset = dataset.ix[:, fixed] 

Аналогично в противоположном случае:

non_fixed = [c for c in dataset.columns if not c.startswith("fixed")] 
dataset.ix[:, non_fixed]