How to add method to existing class in Python?

Question

I have this really convenient high-level pd.DataFrame saving function that I want to add to pandas. How can I add this method to the pd.DataFrame class?

def to_file(df, path, sep="\t", compression="infer", pickled="infer", verbose=False, **args):
    _ , ext = os.path.splitext(path)
    # Serialization
    if pickled == "infer":
        if ext in {".pkl", ".pgz", ".pbz2"}:
            pickled = True
        else:
            pickled = False
    # Compression
    if compression == "infer":
        if pickled:
            if ext == ".pkl":
                compression = None
            if ext == ".pgz":
                compression = "gzip"
            if ext == ".pbz2":
                compression = "bz2"
        else:
            compression = None
            if path.endswith(".gz"):
                compression = "gzip"
            if path.endswith(".bz2"):
                compression = "bz2"
    if verbose:
        print(
            f"path:\t{path}", 
            f"sep:\t{repr(sep)}",
            f"compression:\t{compression}",
            f"pickled:\t{pickled}", 
            sep="\n", 
            file=sys.stderr,
        )
    if pickled == False:
        df.to_csv(path, sep=sep, compression=compression, **args)
    if pickled == True:
        df.to_pickle(path, compression=compression, **args)

no its not he wants to add to the class not to an instance of the class — pippo1980
– pippo1980, Commented Dec 12, 2023 at 18:21

APorter1031 · Accepted Answer · 2018-03-06 20:14:06Z

6

Use python class inheritance. This will allow you to use all the methods in a Pandas data frame and still define your own methods.

import pandas as pd

class NewDF(pd.DataFrame)
    def __init__(self, *args):
        pd.DataFrame.__init__(self, *args)

    def to_file(df, path, sep="\t", compression="infer", pickled="infer", verbose=False, **args):
        ...

answered Mar 6, 2018 at 20:14

APorter1031

2,3166 gold badges19 silver badges40 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

O.rka Over a year ago

pd.DataFrame.to_file = to_file also works but I like the above approach. Thanks, this will come in handy for other things.

Collectives™ on Stack Overflow

How to add method to existing class in Python?

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related