为3D体积绘制Seaborn热图动画时出错 [英] Error while drawing animation of seaborn heatmap for 3D volume

问题描述

尝试显示两个卷之间的互相关， mask_3D ，使用Matplotlib中的 Seaborn heatmap 和 animation 将3D互相关结果可视化为2D图像的渐进动画，但是我遇到了一个错误，请问告诉我如何摆脱这个错误，并正确显示热图?

Trying to visualize the cross-correlation between two volumes, img_3D, and mask_3D, using Seaborn heatmap, and animation from Matplotlib to visualize the 3D cross-correlation result as a progressive animation of 2D images, but I was facing an error, can you please tell me how to get rid of this error, and visualize the heatmaps correctly?

先谢谢了.

Traceback (most recent call last):
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 1705, in __call__
    return self.func(*args)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\_backend_tk.py", line 259, in resize
    self.draw()
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\backend_tkagg.py", line 9, in draw
    super(FigureCanvasTkAgg, self).draw()
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\backend_agg.py", line 392, in draw
    else nullcontext()):
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\contextlib.py", line 112, in __enter__
    return next(self.gen)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backend_bases.py", line 2788, in _wait_cursor_for_draw_cm
    self.set_cursor(self._lastCursor)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\_backend_tk.py", line 544, in set_cursor
    window.configure(cursor=cursord[cursor])
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 1485, in configure
    return self._configure('configure', cnf, kw)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 1476, in _configure
    self.tk.call(_flatten((self._w, cmd)) + self._options(cnf))
_tkinter.TclError: invalid command name "."

使用的代码是:

# Import Libraries
#====================================
import numpy as np
np.random.seed(0)
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
import nibabel as nib
from scipy.signal import correlate

import seaborn as sns
sns.set()
#===================================

img = np.load('img.npy')
act = np.load('act.npy')

# Mode : 'full', 'valid', 'same'

result = correlate(img, act,mode='same')

print(img.shape, act.shape, result.shape)

def updatefig(sl):
    for sl in range(result.shape[2]):
        print(sl,' / ',result.shape[2])
        sns.heatmap(result[...,sl],cbar=False)
        ax.set_title("frame {}".format(sl))
        # Note that using time.sleep does *not* work here!
        plt.pause(0.1)
fig, ax = plt.subplots()

ani = FuncAnimation(fig, updatefig, frames=range(result.shape[2]), interval=5, blit=True)

plt.show()

推荐答案

检查此代码:

import numpy as np
np.random.seed(0)
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
from scipy.signal import correlate
import seaborn as sns
sns.set()

img = np.load('img.npy')
act = np.load('act.npy')

result = correlate(img, act, mode = 'same')

def updatefig(sl):
    ax.cla()
    print(sl + 1, ' / ', result.shape[2])
    sns.heatmap(result[..., sl], cbar = False)
    ax.set_title("frame {}".format(sl + 1))
    ax.axis('off')

fig, ax = plt.subplots()
ani = FuncAnimation(fig, updatefig, frames = result.shape[2], interval = 5)

plt.show()

这给了我这个动画(我将下面报告的动画减半，以将文件大小减小到2 MB以下，上面的代码再现了所有40帧):

which gives me this animation (I halved the animation reported below to reduce the file size under 2 MB, the code above reproduce all 40 frames):

编辑

为了向热图添加固定的颜色条，请检查以下代码:

In order to add a fixed colorbar to the heatmap, check this code:

import numpy as np
np.random.seed(0)
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
from scipy.signal import correlate
import seaborn as sns
sns.set()

img = np.load('img.npy')
act = np.load('act.npy')

result = correlate(img, act, mode = 'same')

def updatefig(sl):
    ax.cla()
    print(sl + 1, ' / ', result.shape[2])
    sns.heatmap(result[..., sl],
                ax = ax,
                cbar = True,
                cbar_ax = cbar_ax,
                vmin = result.min(),
                vmax = result.max())
    ax.set_title("frame {}".format(sl + 1))
    ax.axis('off')

grid_kws = {'width_ratios': (0.9, 0.05), 'wspace': 0.2}
fig, (ax, cbar_ax) = plt.subplots(1, 2, gridspec_kw = grid_kws, figsize = (10, 8))
ani = FuncAnimation(fig, updatefig, frames = result.shape[2], interval = 5)

plt.show()

产生此动画(与上一个相同):

which produces this animation (cut as the previous one):

这篇关于为3D体积绘制Seaborn热图动画时出错的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！