Выбор определенных столбцов данных с обработкой строк

Question

Я пытаюсь выбрать набор столбцов из набора данных, который соответствует определенному условию. Так, например, у меня есть эти столбцы в кадре данных:

['string_category_4', 'string_category_24', 'range_category_6',
  'range_category_17', 'int_numeric_21', 'string_category_15',
  'float_numeric_8', 'int_bool_19', 'int_bool_2']

Каждый из конечных чисел в имени столбца будет уникальным идентификатором, независимо от префикса, например category или numeric. Я хотел бы выбрать все столбцы, которые <=10, поэтому он должен вернуть мне список, который:

['string_category_4', 'range_category_6', 'float_numeric_8',
 'int_bool_2']

Есть ли способ с обработкой строк или даже более простой способ сделать это?

Answer 1

Решение:

columns = [
    'category_4', 'category_24', 'category_6',
    'category_17', 'numeric_21', 'category_15',
    'numeric_8', 'bool_19', 'bool_2'
]
filtered_columns = [col for col in columns if int(col.rsplit('_')[-1]) <= 10]
print(filtered_columns)

Выход:

> ['category_4', 'category_6', 'numeric_8', 'bool_2']