无法使用 matplotlib 将颜色条添加到画布上的热图

Question

我正在尝试将颜色条添加到我在 GUI 中绘制的图形中，但是当代码在没有颜色条代码行的情况下工作时，当我尝试添加颜色条时，我收到一条错误消息“AxesSubplot ' 对象没有属性 'get_array'。我试图找到一个解决方案，但我无法弄清楚我做错了什么。任何帮助将不胜感激。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure
import numpy as np
import tkinter as tk

a = np.array([2,-3,4,-2,1,2,4,3,5])
a = a.reshape(3,3)

class Graph:
    #Draws and return a placeholder for a graph
    #@param parent is the Tk.Frame parent obj
    #@param title is the (string) title for the graph
    def __init__(self, parent, title=''):
        #Initialise graph
        self.title = title
        self.fig = Figure(figsize=(4,4))
        self.plot = self.fig.add_subplot()
        self.plot.set_title(self.title)
        self.plot.set_ylim(top=1)
        self.plot.set_xlim(right=255)
        self.plot.set_ylabel("Certainty")
        self.plot.set_xlabel("Pixel Value")

        #Draw
        self.canvas = FigureCanvasTkAgg(self.fig, master=parent)
        self.canvas.get_tk_widget().pack()
        self.canvas.draw()

        return

    def plotMaxDistHeat(self, maxDistMatrix):
        #Clear previous plot
        self.plot.clear()

        #Plot new results
        self.plot.set_title(self.title)

        self.plot.imshow(maxDistMatrix, cmap='coolwarm')
        self.fig.colorbar(self.plot)

        #Draw Graph
        self.canvas.draw()

root = tk.Tk()

canvas = tk.Canvas(bg='red')
canvas.pack()
graph = Graph(canvas, title='Test Graph')

graph.plotMaxDistHeat(a)

tk.mainloop()

Answer 1

替换

self.fig.colorbar(self.plot)

有

self.fig.colorbar(self.plot.imshow(maxDistMatrix, cmap='coolwarm'))