带有热型重量的 XY 图

Question

我正在做一些迷你项目。我有 x,y 数组，并希望将所有 xy 对与与其权重相对应的数字连接起来（如热图或某事）。位置列表是 X,y 对，而权重列表是数字列表。我的代码如下所示：

def plot_data(layer_num):
    t = list_of_postitions[layer_num]
    c = list_of_weights[layer_num]
    # we create matrix of x,y positions of spots
    r = np.reshape(t,(-1,2))    print(np.shape(r))
    x, y = r.T
    plt.scatter(x,y)
    plt.title('Layer {}'.format(layer_num))
    plt.xlabel("X position")
    plt.ylabel("Y position")
    plt.show()

谢谢您的帮助

Answer 1

根据pyplot 文档，您可以使用s参数来更改每个点的大小（如果您希望绘图大小对应于权重）或c参数来调整颜色（有说明如何使用它的教程，这里有一个示例）。s和参数都可以是位置和c长度相同的数组。xy