How to skip coloring inf values?-CodePudding

I'm styling some data in excel after processing in pandas. But I have some bad data that gives me lot of inf values. Is there a way to skip coloring inf values?

        cr = sns.light_palette("green", as_cmap=True)
        styler = data.style
        zeleno = styler.background_gradient(
            cmap=cr,
            subset=data.columns.get_loc_level('value1', level=1)[0]
        )
        styler.to_excel(writer, sheet_name=name)

I got data like this.

I need to be like this, (skip inf values, they can be white or some other colors)

CodePudding user response：

Currently, it is still a better approach to use

*Styling only applied to subset (value1 on level=1)

Setup and Imports Used:

import numpy as np
import pandas as pd  # version 1.3.3
import seaborn as sns
from matplotlib import colors as mcolors

np.random.seed(25)
a = np.random.random((3, 4))
# Replace Randomly With inf
a[np.random.randint(0, 2, size=a.shape) == 0] = np.inf

data = pd.DataFrame(a, columns=pd.MultiIndex.from_product([
    list('AB'),
    ['value1', 'value2']
]))
data