Просто перейдите по DataFrame.columns
, теперь это пример, в котором вы получите список имен столбцов, которые соответствуют:
import pandas as pd
data = {'spike-2': [1,2,3], 'hey spke': [4,5,6], 'spiked-in': [7,8,9], 'no': [10,11,12]}
df = pd.DataFrame(data)
spike_cols = [col for col in df.columns if 'spike' in col]
print(list(df.columns))
print(spike_cols)
Выход:
['hey spke', 'no', 'spike-2', 'spiked-in']
['spike-2', 'spiked-in']
Объяснение:
df.columns
возвращает список имен столбцов [col for col in df.columns if 'spike' in col]
выполняет итерацию по списку df.columns
с переменной col
и добавляет ее в результирующий список, если col
содержит 'spike'
. Этот синтаксис является пониманием списка . Если вы хотите, чтобы результирующий набор данных с соответствующими столбцами, вы можете сделать это:
df2 = df.filter(regex='spike')
print(df2)
Выход:
spike-2 spiked-in
0 1 7
1 2 8
2 3 9