我正在尝试使用 matplotlib 生成热图。
import matplotlib.pyplot as plt
import numpy as np
import sys
import re
data = np.random.rand(10, 10)
heatmap = plt.pcolor(data)
with open(sys.argv[1],'r') as iFile:
for line in iFile:
line=line.strip()
values=re.split('[-:]',line)
x=values[0].strip()
y=values[1].strip()
z=values[2].strip()
data[y,x]=z
for y in range(data.shape[0]):
for x in range(data.shape[1]):
plt.text(x + 0.5, y + 0.5, '%.2f' % data[y, x],
horizontalalignment='center',
verticalalignment='center',)
plt.colorbar(heatmap)
plt.show()
并且数据采用“x - y : value”格式如下
6 - 4 : 0.180671274019055
6 - 1 : 0.30475569499109
6 - 3 : 0.276460025706412
6 - 2 : 0.298002584369681
...
...
1 - 4 : 0.0961071651182192
1 - 1 : 0.259655770346209
1 - 3 : 0.308485173534571
1 - 2 : 0.278724535194018
问题：x 和 y 值是非常动态的并且可以变化（这里是 1 到 6）。我想根据输入文件动态设置轴。任何帮助将不胜感激，谢谢！