请教一个 Python pandas dataframe 特定列单元格内容高亮的问题

import pandas as pd
import os
import webbrowser

def write_data_to_file(filename, content):

    file = open(filename, mode='w')
    file.write(content)
    file.close()


def highlight_hot_color(col):

    hot_color = ['red', 'yellow', 'orange']
    check = [item in hot_color for item in col]
    return ['background-color: yellow' if v else '' for v in check]

if __name__ == '__main__':
    df = pd.DataFrame([['Allen', 'red', 20], ['Tom', 'yellow', 30], ['Jack', 'blue', 40], ['Bob', 'grey', 50]], 
        columns=['name', 'color', 'age'])
    df.style.apply(highlight_hot_color, subset=['color'])

    html = df.to_html(index=False)
    file_name = 'test.html'
    path_name = os.path.abspath(file_name)
    url = 'file://' + path_name
    write_data_to_file(path_name, html)
    webbrowser.open(url)

dataframe

HTML

yellow'

import

5 条回复 • 2021-03-05 17:13:04 +08:00

ZAXON

2021-03-04 09:49:57 +08:00

```python
df = pd.DataFrame([['Allen', 'red', 20], ['Tom', 'yellow', 30], ['Jack', 'blue', 40], ['Bob', 'grey', 50]],
columns=['name', 'color', 'age'])
df_result = df.style.apply(highlight_hot_color, subset=['color'])

with open('test.html','w') as f:
f.write(df_result.render())
```

试试这样写？最后的生成 html 我是出于方便这么写的，你可以再看看有没有更好的写法。

我更好奇的是第一个需求，是要达到 https://github.com/pandas-dev/pandas/issues/18066#issuecomment-371150717 这样的结果吗

hackpro

2021-03-04 12:18:00 +08:00

@ZAXON 感谢大佬回复

1 、高亮的问题部分解决了但有一个问题
df.style.apply 返回的是一个 Styler 类型不再是 dataframe
如果我需要给 dataframe 用上 pretty_html_table.build_table 进行表格风格挑战的话还是很麻烦

2 、您提到的第一个问题我是想做成链接里面的那样但是好像难度很大
目前输出到命令行我用的就是最原始的 print(df.to_string()) 😄

ZAXON

2021-03-04 14:25:47 +08:00

个人感觉 dataframe -> html.table 最灵活的方法还是直接使用 jinja2，我在公司邮件里面生成表格就是这样操作的。

terminal 的格式化输出我推荐用的 https://github.com/willmcgugan/rich 这个库，不过不确定能不能解决你的这个问题。

TimePPT

2021-03-04 14:31:25 +08:00

@ZAXON jinja2+1，如果长期有样式需求，最好业务和样式分开处理。

hackpro

2021-03-05 17:13:04 +08:00

@ZAXON @TimePPT 感谢两位老哥的建议和解答

我现在的解决办法是用 dataframe.style 解决表格输出时高亮的问题，但是默认的表格实在是太丑了
所以找了个 pretty_html_table 的包，可以非常方便的解决表格样式的问题
from pretty_html_table import build_table
html = build_table(data_frame, 'green_light')
write_data_to_file('test.html', html)

但是 pretty_html_table.build_table()接受 dataframe 作为参数，输出的是 str
而我们之前讨论的 style.apply()接受 dataframe 作为参数，输出的是 pandas.io.formats.style.Styler
两者不兼容有什么办法把这两个结合在一起吗😭