Сравните два столбца во фрейме данных и выделите значение, используя стиль панд - PullRequest
Новая
       26

Сравните два столбца во фрейме данных и выделите значение, используя стиль панд

0 голосов
/ 18 декабря 2018

Я пытаюсь выделить некоторые значения в некоторых столбцах во фрейме данных с помощью стилей панд, таких как: 

import pandas as pd
import numpy as np

np.random.seed(24)
df = pd.DataFrame({'A': np.linspace(1, 10, 10)})
df = pd.concat([df, pd.DataFrame(np.random.randn(10, 4), 
columns=list('BCDE'))],axis=1)
df.iloc[0, 2] = np.nan

def highlight_greater(row):

    color=""
    if row['B'] > row['C']:
       color = 'red'
    elif row['D'] > row['E']:
        color = 'gray'

    background = ['background-color: {}'.format(color) for _ in row]
    return background

with open ('out.html','w') as out:
    print >> out, df.style.apply(highlight_greater, axis=1).render()

Это нормально работает, но не соответствует моему объекту, я хочу только выделить B и Dколонны.Этот сценарий выделяет все столбцы в строке, если они соответствуют условию.Любая идея ?Спасибо

1 Ответ

0 голосов
/ 18 декабря 2018

Вы можете изменить пользовательскую функцию для стилей данных: 

def highlight_greater(x):
    r = 'red'
    g = 'gray'

    m1 = x['B'] > x['C']
    m2 = x['D'] > x['E']

    df1 = pd.DataFrame('background-color: ', index=x.index, columns=x.columns)
    #rewrite values by boolean masks
    df1['B'] = np.where(m1, 'background-color: {}'.format(r), df1['B'])
    df1['D'] = np.where(m2, 'background-color: {}'.format(g), df1['D'])
    return df1


df.style.apply(highlight_greater, axis=None)
...