Для цикла для удаления строкового шаблона из имени столбца

Question

Я пытаюсь удалить '_Adj' из имени столбца в кадре данных 'df_merged', если (1) имя столбца содержит «eTIV» или «eTIV1».

for col in df_merged.columns:
    if 'eTIV1' in col or 'eTIV' in col:
        df_merged.columns.str.replace('_Adj', '')

Этот кодкажется, выдает следующую ошибку:

KeyError: '[] не найден по оси'

Answer 1

Вот два варианта:

Вариант 1

df_merged.columns = [col.replace('_Adj','') if 'eTIV' in col else col for col in list(df_merged.columns)]

Вариант 2

df_merged = df_merged.rename(columns={col: col.replace('_Adj','') if 'eTIV' in col else col for col in df_merged.columns})