javascript - Python Pandas 突出显示 Dataframe 中的行

Question

我有一个返回结果表的 mysql 查询。 如下图:

---

我想要实现的是在最后一列的值为“MM”时突出显示整行。

我过去使用过一小段 JS 代码，它只适用于单个单元格，而不适用于整行。

---

我的代码是这样的:

import pandas as pd
from pandas import DataFrame
from pandas.tseries.offsets import BDay
import sys
import mysql.connector
import time
import datetime
from datetime import date,timedelta
import os


## testing dates
currentdate = pd.datetime.today()
today = currentdate.strftime("%Y-%m-%d")
yesterday = currentdate - BDay(1)
yest = yesterday.strftime("%Y-%m-%d")
#print today
#print yest

## where to dump the file
develop_path = "/var/www/html/exDiv/"
report_file = "exDivReport." + today + ".html"

## function to get data and print out
def get_data_from_hk():
    cnx = mysql.connector.connect(
        user='some_user', 
        password='some_pass',
        host='some_host',
        database='some_database'
    )

    cursor = cnx.cursor()
    query = ("SELECT c.description, c.feedcode, date(exdividenddate), dividend, (case when c.metadata like '%3=MM%' then 'MM' when c.metadata is NULL then 'NM' when c.metadata not like '%3=MM%' then 'NM' else c.metadata end) as metadata FROM dividends d join contracts c using(contracttag) where d.kind=0 and d.exdividenddate=getNextTradingDate(1, curdate());")

    cursor.execute(query)
    dbdata_hk=cursor.fetchall()
    cnx.close()
    print dbdata_hk
    return dbdata_hk


def generate_report():
    report_colnames = ['description', 'feedcode', 'date(exdividenddate)', 'dividend', 'metadata'] 

    dbdata_list_hk = get_data_from_hk()
    df = pd.DataFrame(dbdata_list_hk, columns=report_colnames)

    ##test: if the last column contains 'MM' then highlight table row
    ##highlight = lambda x:  '<span class="market_make">{:,.0f}</span>'.format(x) if x = 'MM' else '{:,.0f}'.format(x)

    with open(os.path.join(develop_path,report_file), 'w') as f:
        f.write("<b>Stocks with ex-dividend on next trading day<b/><br/><br/>")
        f.write(df.to_html(
            index=False,
            ##the whole row needs to be highlighted
            ##formatters={('description', 'metadata'): highlight}, 
            escape=False,
            justify='center'
        ))
        f.write(style_text)

    return df

## colouring js
style_text = '''
<style>
.market_make {
    background-color: #FBB117;
}
</style>
<script>
var elements = document.getElementsByClassName('market_make');

for(var i=0; i<elements.length; i++) {
    elements[i].parentElement.setAttribute("bgcolor", "#FBB117");
}
</script>
'''

# run the report
d=generate_report()
print d

---

pandas 创建的 html 部分是这样的:

<b>Stocks with ex-dividend on next trading day<b/><br/><br/><table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th>description</th>
      <th>feedcode</th>
      <th>date(exdividenddate)</th>
      <th>dividend</th>
      <th>metadata</th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <td>HHI Div</td>
      <td>HHI</td>
      <td>2015-07-02</td>
      <td>34.793976</td>
      <td>NM</td>
    </tr>
    <tr>
      <td>CCB</td>
      <td>939</td>
      <td>2015-07-02</td>
      <td>0.000000</td>
      <td>MM</td>
    </tr>
    <tr>
      <td>null Combo + Stock - Legs</td>
      <td>FXI-H</td>
      <td>2015-07-02</td>
      <td>35.327749</td>
      <td>NM</td>
    </tr>
    <tr>
      <td>CSE</td>
      <td>1088</td>
      <td>2015-07-02</td>
      <td>0.000000</td>
      <td>MM</td>
    </tr>
    <tr>
      <td>PIN</td>
      <td>1339</td>
      <td>2015-07-02</td>
      <td>0.011800</td>
      <td>NM</td>
    </tr>

---

除了 JS 之外，还有更有效的方法吗？

我可以将它与一些 CSS 结合起来吗？

有什么想法吗？

Answer 1

这可以通过 pd.DataFrame.style.apply 和自定义样式函数来完成!

例如，要将“MM”行突出显示为黄色:

def custom_style(row):

    color = 'white'
    if row.values[-1] == 'MM':
        color = 'yellow'

    return ['background-color: %s' % color]*len(row.values)

df.style.apply(custom_style, axis=1)