在Python中创建信息图

Question

我想在python中创建一个简单的信息图。 Matplotlib似乎有很多功能，但没有什么能覆盖我简单的热图网格示例。

信息图是一个简单的5 x 5网格，数字范围从0到1.然后网格方块将被着色为0 =白色1 =蓝色0.5是淡蓝色。

可能会使用Matplotlib，但我无法找到或合并任何能够深入了解这一点的示例。

任何见解，示例代码或库方向都会有所帮助

此致 马特

Answer 1

这取决于您拥有图表后需要做什么，Matplotlib允许您以交互方式在屏幕上显示图表，将其保存为矢量，pdf或位图格式等等。

如果你选择这个框架，imshow会做你需要的，这是一个例子：

# Just some data to test:
from random import gauss
a = [[gauss(0, 10) for i in xrange(0, 5)] for j in xrange(0,5)]

from pylab import * # or just launch "IPython -pylab" from the command line

# We create a custom colormap:
myblue = cm.colors.LinearSegmentedColormap("myblue", {
    'red':   [(0, 1, 1), (1, 0, 0)], 
    'green': [(0, 1, 1), (1, 0, 0)],
    'blue':  [(0, 1, 1), (1, 1, 1)]})

# Plotting the graph:
imshow(a, cmap=myblue)

有关色彩地图check this link的详细信息，此处为link for imshow - 或只是使用help(colors.LinearSegmentedColormap)和help(imshow)。

alt text http://img522.imageshack.us/img522/6230/bluep.png

（请注意，这是标准选项的结果，您可以添加网格，更改过滤等）。

---

修改

  然而，我正在寻找显示   网格中的数字

为了简单起见：

for i in xrange(0,5):
    for j in xrange(0,5):
        text(i, j,
             "{0:5.2f}".format(a[i][j]),
             horizontalalignment="center",
             verticalalignment="center")

Answer 2

PyCairo是你的朋友。简单的例子：

from __future__ import with_statement
import cairo
img = cairo.ImageSurface(cairo.FORMAT_ARGB32,100,100)
g = cairo.Context(img)
for x in range(0,100,10):
    for y in range(0,100,10):
        g.set_source_rgb(.1 + x/100.0, 0, .1 + y/100.0)
        g.rectangle(x,y,10,10)
        g.fill()
with open('test.png','wb') as f:
    img.write_to_png(f)

您可能会发现this tutorial有帮助。

Answer 3

一种可能性是从python生成SVG。您可以在Firefox或Inkscape中查看SVG。

这是一个快速而又肮脏的例子：

import random

def square(x, y, value):
    r, g, b = value * 255, value * 255, 255
    s = '<rect x="%d" y="%d" width="1" height="1" style="fill:rgb(%d,%d,%d);"/>' % (x, y, r, g, b)
    t = '<text x="%d" y="%d" font-size=".2" fill="yellow">%f</text>' % (x, y + 1, value)
    return s + '\n' + t

print('''
<?xml version="1.0" standalone="no"?>
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN"
"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">

<svg width="100%" height="100%" version="1.1" viewBox="0 0 5 5"
xmlns="http://www.w3.org/2000/svg">
''')
for x in range(0, 5):
    for y in range(0, 5):
        print(square(x, y, random.random()))

print('</svg>')

alt text http://www.imagechicken.com/uploads/1257184721026098800.png