I just created a wrapper on top of pandas for this purpose, if it helps anyone :)
You can achieve it this way:

pip install pandoras

import pandoras as pd
df = pd.DataFrame({"A": [1, 2, 3], "B": [4, None, 6]})
df.drop(columns=["B"], inplace=True). # or edit,apply,...
df.undo()
df.redo()

This is the repo for anyone interested in contributing:
https://github.com/al2m4n/pandoras