Раскраска кадра данных перед сохранением - PullRequest
0 голосов
/ 16 октября 2018

Я создал df, где значениями являются * "Да", "Нет" и"Возможно". *

Я хотел бы сделать эти ячейки красочными перед сохранением в Excel,Если я использую библиотеку xlwt , у меня будет что-то вроде этого:

sheet2.write(row, col, str('Allowed'), style)

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

Мой фрейм данных выглядит примерно так:

enter image description here

Мой желаемый вывод:

enter image description here

Любые предложения о том, как это сделать?

1 Ответ

0 голосов
/ 16 октября 2018

как то так?

def color_df(val): 
    if val == 'Yes':
        color = 'red'
    elif val == 'Maybe':
        color = 'Orange'
    else:
        color = 'Blue'
    return 'color: %s' % color

df = df.style.applymap(color_df).to_excel(path_var + 'styled.xlsx', engine='xlsxwriter')
...