我有一个 Pandas 数据帧,希望将其作为 matplotlib 表格绘制出来。到目前为止,我已经使用以下代码成功实现了这一部分:
import numpy as np
randn = np.random.randn
from pandas import *
idx = Index(arange(1,11))
df = DataFrame(randn(10, 5), index=idx, columns=['A', 'B', 'C', 'D', 'E'])
vals = np.around(df.values,2)
fig = plt.figure(figsize=(15,8))
ax = fig.add_subplot(111, frameon=True, xticks=[], yticks=[])
the_table=plt.table(cellText=vals, rowLabels=df.index, colLabels=df.columns,
colWidths = [0.03]*vals.shape[1], loc='center')
table_props = the_table.properties()
table_cells = table_props['child_artists']
clm = cm.hot(vals)
for cell in table_cells:
cell.set_height(0.04)
# now i would like to set the backgroundcolor of the cell
在最后,我想根据着色图设置单元格的背景颜色 - 但是如果没有索引,我该如何在clm数组中查找它?
另一个问题:我能以某种方式向表格传递格式化字符串,以便将文本格式化为2位小数吗?
欢迎任何提示, Andy