Как правильно переопределить (или создать свой собственный) метод Pandas read_csv?

Question

Я пытаюсь написать свой собственный метод read_csv поверх pandas .read_csv.

# pandas_custom.py module
import pandas as pd
def read_csv(**kwargs):
    df = pd.read_csv(**kwargs)
    df = df.apply(my_function)
    return df

# main_module.py module
import pandas_custom as pdF
df = pdF.read_csv(filepath_or_buffer='train.csv') # WORKS
df = pdF.read_csv('train.csv') # TypeError: read_csv() takes 0 positional arguments but 1 was given

Кто может мне правильно создать свой собственный метод read_Csv или переопределить оригинал? Должен ли я просто скопировать все параметры из https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.read_csv.html и вставить их в мой метод read_csv?

Answer 1

Вы пытаетесь передать аргумент своей функции, но ваша функция принимает только kwargs. Попробуйте:

def read_csv(*args, **kwargs):
    df = pd.read_csv(*args, **kwargs)
    df = df.apply(my_function)
    return df