循环内的交互式Matplotlib 3d散点图

Question

我有一个关于在循环内使用pyhton和matplotlib进行3d散点图的交互式可视化的问题。要绘制的点由循环内部的函数生成，因为它们分别应用于多个数据集。我的想法是将新点添加到绘图中，因为它们是在每次迭代中生成的。我正在尝试这样的事情：

import matplotlib.pyplot as plt    

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
plt.ion()

for i in range(N):
    res=function(data_in[i])
    ax.scatter(res[0],res[1],res[2])
    plt.pause(0.001)

一切似乎都可以正常工作，但是问题在于生成点的函数相当慢，并且在工作时图变为静态。生成新点时，无法旋转图，缩放图，最大化图等。

有没有一种方法可以在功能正常工作时与图进行交互？