matplotlib: sử dụng bản đồ màu để tô màu nền bảng

Question

Tôi có một khung dữ liệu Pandas và tôi muốn vẽ nó thành bảng matplotlib. Cho đến nay tôi đã phần đó làm việc với mã sau:

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 

Vào cuối này tôi muốn thiết lập background-color của tế bào theo colormap - nhưng làm thế nào để tôi nhìn nó lên trong CLM mảng không có chỉ mục?

Một câu hỏi khác: tôi có thể bằng cách nào đó chuyển chuỗi định dạng vào bảng để định dạng văn bản thành 2 chữ số thập phân không?

Bất kỳ gợi ý đánh giá cao, Andy

Answer 1

Bạn có thể sử dụng plt.normalize() để tạo một đối tượng Normalize để chuẩn hóa dữ liệu của bạn, và thông qua các dữ liệu bình thường hóa đến đối tượng bản đồ màu để có được màu sắc.

plt.table() có đối số cellColours đặt màu nền của mỗi ô.

Bởi vì bản đồ màu cm.hot sử dụng màu đen cho giá trị tối thiểu, tôi đã tăng phạm vi giá trị khi tạo đối tượng chuẩn hóa.

Đây là mã:

from matplotlib import pyplot as plt 
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) 
normal = plt.normalize(vals.min()-1, vals.max()+1) 

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', 
        cellColours=plt.cm.hot(normal(vals))) 

Answer 2

mã làm việc của Andy:

#!/usr/bin/env python 
# -*- coding: utf-8 -*- 

# sudo apt-get install python-pandas 
# sudo apt-get install python-matplotlib 
# 
# python teste.py 

from matplotlib import pyplot 
from matplotlib import cm 

import numpy 

from pandas import * 

idx = Index(numpy.arange(1, 11)) 

df = DataFrame(
     numpy.random.randn(10, 5), 
     index=idx, 
     columns=['A', 'B', 'C', 'D', 'E'] 
    ) 

vals = numpy.around(df.values, 2) 

normal = pyplot.normalize(vals.min()-1, vals.max()+1) 

fig = pyplot.figure(figsize=(15, 8)) 

ax = fig.add_subplot(111, frameon=True, xticks=[], yticks=[]) 

the_table = pyplot.table(
       cellText=vals, 
       rowLabels=df.index, 
       colLabels=df.columns, 
       colWidths = [0.03]*vals.shape[1], 
       loc='center', 
       cellColours=pyplot.cm.hot(normal(vals)) 
      ) 

pyplot.show()