How to color specific cells given row index and column name in pandas?

Question content

I have a dataframe df of shape (100,10) and a dictionary columns_to_color for each row.

columns_to_color = {
  0: ['col1', 'col2', 'col9'],  # Columns 1, 2 and 9 should be colored for row 0
  1: ['col1', 'col5', 'col8'],  # Columns 1, 5 and 8 should be colored for row 1
  2: ['col3', 'col4', 'col7'],  # Columns 3, 4 and 7 should be colored for row 2
  .......

How to color specific cells for each row of this column? And then save to excel?

Correct answer

You can use . style.apply() Like this:

df.style.apply(
    lambda row: [
        'font-weight: bold' if col in columns_to_color[row.name] else ''
        for col in row.index],
    axis=1)

(I use bold only for convenience.)

Then to convert to excel, use .to_excel().

Example:

df = pd.DataFrame({f'col{n}': range(10*n, 10*n + 3) for n in range(10)})

After apply (in my ide):

The above is the detailed content of How to color specific cells given row index and column name in pandas?. For more information, please follow other related articles on the PHP Chinese website!