使用R或Python加热二进制数据图

Question

我有一个0和1的二进制数据集,其中0表示缺席,1表示存在事件.

数据集的示例如下所示:

events    germany    Italy 
Rain      0          1
hail      1          0
sunny     0          0

我想通过从文件中读取数据,以热图的形式获得这些数据的红白图片.

Answer 1

编辑:响应下面的评论,这是一个示例数据文件(在磁盘上保存为"data.txt"):

Rain  0 0 0 0 1 0 1 0 0 1
Hail  0 1 0 0 0 0 0 1 0 0
Sunny 1 1 1 0 1 0 1 0 1 1

在python中,我们可以通过以下方式读取标签并绘制此"热图":

from numpy import loadtxt
import pylab as plt

labels = loadtxt("data.txt", usecols=[0,],dtype=str)
A      = loadtxt("data.txt", usecols=range(1,10))

plt.imshow(A, interpolation='nearest', cmap=plt.cm.Reds)
plt.yticks(range(A.shape[0]), labels)

plt.show()
import pylab as plt