Создайте подмножество DataFrame, зависящее от имени столбца

Question

У меня есть файл данных pandas под названием timedata с разными именами столбцов, некоторые из которых содержат слово Вибрация, некоторые эксцентриситеты. Можно ли создать кадр данных только столбцов, содержащих слово Вибрация?

Я попытался с помощью

vib=[] 
for i in timedata: 
    if 'Vibration' in i: 
     vib=vib.append(i) 

, чтобы затем создать DataFrame на основе indicies этих столбцов. Это действительно не самый эффективный способ сделать это, и я уверен, что должно быть что-то простое в понимании списка.

EDIT

Dataframe формы:

df = DataFrame({'Ch 1:Load': randn(10), 'Ch 2:Vibration Brg 1T ': randn(10), 'Ch 3:Eccentricity Brg 1H ': randn(10), 'Ch 4:Vibration Brg 2T ': randn(10)}) 

К сожалению у меня медленный день! спасибо за любую помощь

Answer 1

Что-то вроде этого, чтобы вручную выбрать все столбцов со словом «Вибрация» в нем:

df[[col for col in df.columns if "Vibration" in col]] 

Вы также можете сделать то же самое с помощью метода filter:

df.filter(like="Vibration") 

Если вы хотите сделать более гибкий фильтр, вы можете использовать опцию regex. Например. посмотреть, если «Вибрация» или «Еккл» в имени столбца:

df.filter(regex='Ecc|Vibration') 

Answer 2

newDf = Df.loc[:,['Vibration']] 

или

newDf = Df.loc[:,['Vibration','eccentricity']] 

, чтобы получить больше collumns

для поиска значения в collumn:

newDf = Df[Df["CollumnName"] == "vibration"]