Как сохранить определенные столбцы при фильтрации данных по регулярным выражениям

Question

Я предварительно формирую фильтр для столбцов, содержащих tweet_volume, используя регулярное выражение, с помощью следующей команды:

tweet_volume_df = df_merged.filter(regex=("tweet_volume.*"))

В пределах df_merged есть столбец для name.Я хотел бы включить столбцы name в tweet_volume_df.

Как выполнить эту операцию в одной строке?

Answer 1

Вы можете сделать с

df.filter(regex=r'(Name|tweet_volume.*)',axis=1)

Answer 2

Вы можете использовать "|"в регулярном выражении, чтобы соответствовать name или tweet_volume. *:

import pandas as pd

data = [list(range(5))]
headers = ['name', 'tweet_volume1', 'tweet_volume2', 'a', 'b']
df_merged = pd.DataFrame(data, columns=headers)

tweet_volume_df = df_merged.filter(regex=("tweet_volume.*|name"))