Удалите все пробелы из заголовков кадра данных Polars. ⇐ Python

[Anonymous]

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

Код: Выделить всё

import polars as pl

csv_file = b'''
A   \tB    \tC   \tD \t  E
CD\tE\t300 0\t0\t0
CD\tE\t1071 0\t0\t0
K\tE\t390 0\t0\t0
'''.strip()

Я хочу прочитать файл, а затем удалить все пробелы и/или табуляции из имен столбцов. В настоящее время да

Код: Выделить всё

df = pl.read_csv(csv_file,
comment_prefix='#',
separator='\t')
df = df.rename(lambda column_name: column_name.strip())

Это «полярный» способ сделать это? Я не большой поклонник лямбда-выражений, но если единственное другое решение — написать функцию только для этого, думаю, я буду придерживаться лямбда-выражений.


Подробнее здесь: https://stackoverflow.com/questions/783 ... -dataframe