如何使用matplotlib

Question

我已经将我的数据标准化为一个颜色图，我只是不知道如何绘制它。这是我要创建的图形类型：

我想使用cmap'bwr'创建类似的东西，其中值'1'将为纯红色，而值'0'将为纯蓝色。我用它来规范化我的数据并将其映射到颜色：

norm = Normalize(vmin = min(data), vmax = max(data), clip = True)
mapper = cm.ScalarMappable(norm = norm, cmap = plt.get_cmap('bwr'))

因此data[0]将位于图形的最左侧，列表中的最后一个值将位于最右侧。

Answer 1

如果我理解正确，则要创建图像。可以通过imshow完成。

import numpy as np
import matplotlib.pyplot as plt

# some data
data = np.atleast_2d(np.sin(np.sqrt(np.linspace(1,250,250)))*34)
# plot image
plt.imshow(data, aspect="auto", cmap="bwr")
plt.show()