Błąd "can't set attribute" przy zmianie wartości DataFrame.style

def highlight_max(x, color):
    return np.where(x == np.nanmax(x.to_numpy()), f"color: {color};", None)
df = pd.DataFrame(np.random.randn(5, 2), columns=["A", "B"])
df.style.apply(highlight_max, subset = ["A"], color='red', axis = 0)  
df.style.apply(highlight_max, subset = ["B"], color='blue', axis=0)  

def highlight_max(x, color):
    return np.where(x == np.nanmax(x.to_numpy()), f"color: {color};", None)

df = pd.DataFrame(np.random.randn(5, 2), columns=["A", "B"])
df.style.apply(highlight_max, subset = ["A"], color='red', axis = 0).apply(highlight_max, subset = ["B"], color='blue', axis=0)  

def highlight_max(x, color):
    return np.where(x == np.nanmax(x.to_numpy()), f"color: {color};", None)

df = pd.DataFrame(np.random.randn(5, 2), columns=["A", "B"])

styler = df.style.apply(highlight_max, subset = ["A"], color='red', axis = 0)\
.apply(highlight_max, subset = ["B"], color='blue', axis=0)  


# nowy DataFrame, ale korzystamy ze stylu poprzedniego
df2 = pd.DataFrame(np.random.randn(5, 2), columns=["A", "B"])
styler2 = df2.style.use( styler.export())


styler2 

styler2.data

# Data
df = pd.DataFrame(np.random.randn(5, 2), columns=["A", "B"])

# Pierwsza funkcja ktora koloruje kolumne "A"
def f1(df):
    # Najprosciej by bylo gdyby dalo sie zrobic cos takiego ale nie sie da:
    # df.style = df.style.apply(highlight_max, subset = ["A"], color='red', axis = 0)
    
    # Wiec tworze styler
    styler = df.style
    styler = styler.apply(highlight_max, subset = ["A"], color='red', axis = 0)
    # Chce zeby funkcja zwrocila df ale ze zmienonym stylem. W tym momencie styler to odrebny obiekt. 
    # Niestety jak zwroce styler.data to dataframe nie ma tego stylu ktory wprowadzilem
    # return styler.data
    
    # Funkcje export i use nic mi nie daja bo wciaz jedyne co chce zwrocic to df ze zminionym stylem
    # To zwraca styl, a nie dataframe ze zmienionym stylem
    return df.style.use(styler.export())
    
    # Wydaje mi sie ze moj cel jest prosty i dziwi mnie ze jest to takie trudne 
    
# Druga funkcja ktora koloruje kolumne "B"
def f2(df):
    df.style.apply(highlight_max, subset = ["B"], color='blue', axis = 0)
    return df

res = f1(df)
type(res)

df = pd.DataFrame(np.random.randn(5, 2), columns=["A", "B"])

def f1(styler):
    return styler.apply(highlight_max, subset = ["A"], color='red', axis = 0)
    
def f2(styler):
    return styler.apply(highlight_max, subset = ["B"], color='blue', axis = 0)



style = df.style
f1(style)
f2(style)
style

# ale poniższe pokazuje dane już bez kolorowania bo został utworzony nowy "pusty" obiekt Styler, a nie ten teoretycznie "zapamiętany"
df.style