Есть ли способ обрезать / вырезать пробелы в нескольких столбцах кадра данных панд?

Question

У меня есть пандас с 5 столбцами, и 3 из этих столбцов являются строковыми столбцами. Я хочу обрезать все начальные и конечные пробелы в этих 3 столбцах. Есть ли способ достичь этого за один раз?

  ID    col_1  col_2    col_3   col_4
0  1      AA     XX      200     PP
1  2      BB     YY      300     QQ
2  3      CC     ZZ      500     RR

Я хочу обрезать 'col_1', 'col_2', 'col_4'

Я знаю, df['col_1'].str.strip() работает на отдельном столбце. Но могу ли я сделать все столбцы за один раз?

Answer 1

Используйте DataFrame.apply со списком столбцов:

cols = ['col_1', 'col_2', 'col_4']
df[cols] = df[cols].apply(lambda x: x.str.strip())

Или анализируйте только столбцы объекта, это очевидно строки:

cols = df.select_dtypes(object).columns
df[cols] = df[cols].apply(lambda x: x.str.strip())