python - 表格单元格2或3比例着色的条件格式

我想将一个简单的表格输出到PDF文件，并根据其值对单元格进行2或3比例着色的一些条件格式设置。类似于Microsoft Excel条件格式设置选项中的红白绿颜色缩放。

import pandas
import numpy as np
df = pandas.DataFrame(np.random.randn(10, 2), columns=list('ab'))
print df

#Output:
          a         b
0 -1.625192 -0.949186
1 -0.089884  0.825922
2  2.117651 -0.046258
3 -0.921751 -0.144447
4 -0.294095 -1.774725
5 -0.780523 -0.435909
6  0.544958  0.303268
7  0.014335  0.036182
8 -0.756565  0.120711
9  1.145055  0.542755

现在，我想将其输出到具有a列且独立于b列的3比例条件格式的表中的PDF中，因此我的输出将类似于下面的Excel示例。

有点像，但按列:

最佳答案

好的，我做了一些研究，这几乎涵盖了我想要做的事情。使用matplotlib颜色图可以解决问题。这是 map 颜色选项的链接。
matplotlib color maps

我使用了RdYlGn着色。

import pandas as pd
import numpy as np
from matplotlib import pyplot as plt

df = pd.DataFrame(np.random.randn(10, 8), columns=list('abcdefgh'))

print df

#Round to two digits to print nicely
vals = np.around(df.values, 2)
#Normalize data to [0, 1] range for color mapping below
normal = (df - df.min()) / (df.max() - df.min())

fig = plt.figure()
ax = fig.add_subplot(111)
ax.axis('off')
the_table=ax.table(cellText=vals, rowLabels=df.index, colLabels=df.columns,
                   loc='center', cellColours=plt.cm.RdYlGn(normal),animated=True)
fig.savefig("table.png")

输出: