matplotlib热图，具有自定义颜色的离散值

Question

我试图根据Python中的值为每个单元格生成一个包含自定义颜色的热图。

var express = require('express');
var server = express();
var engines = require('consolidate');

server.set('views',__dirname);
server.engine('html',engines.mustache);
server.set('view engine','html')
server.use(express.static(__dirname));

server.get('/:name',function(req,res){
    res.render('index');
});

server.listen(3000,function(){
    console.log("Listening on localhost:3000");
})

任何人都可以提供帮助吗？

Answer 1

这是一个工作的MWE：

from matplotlib import colors
data = array([[1,2,3],[2,3,5], [3,1,2]])
cols = {1:'red',2:'blue',3:'green',5:'purple'}

cvr = colors.ColorConverter()
tmp = sorted(cols.keys())
cols_rgb = [cvr.to_rgb(cols[k]) for k in tmp]
intervals = array(tmp + [tmp[-1]+1]) - 0.5
cmap, norm = colors.from_levels_and_colors(intervals,cols_rgb)
plt.pcolor(data,cmap = cmap, norm = norm)

结果如下：