matplotlib的三维点图不能填充到整个画布

我在画matplotlib的三维点图的时候遇到了输出的图片太小的问题，如下：

x=passrate_matrix3d['basic_com']
y=passrate_matrix3d['pre_com']
z=passrate_matrix3d['head']
v=passrate_matrix3d['passrate']
min_v = min(v)
max_v = max(v)
color = [plt.get_cmap("bwr", 100)(int(float(i-min_v)/(max_v-min_v)*100)) for i in v]
fig = plt.figure(figsize=(6,6))
ax = fig.add_subplot(111,projection='3d')
plt.set_cmap(plt.get_cmap("bwr", 100))
ax.get_proj = lambda: np.dot(Axes3D.get_proj(ax), np.diag([0.5, 0.5, 1, 1]))
im = ax.scatter(x, y, z, s=10,c=color,marker='.')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
plt.show()

我在 jupyter lab 下运行的代码，加载了一个matplotlib的可视化交互插件ipympl，passrate_matrix3d是数据表格。现在图的部分和画布不成比例，不能进一步观察图的细节。我试着更改了轴的比例和整个画布的大小，都不行。想请教一下有没有办法让真正图的部分大一些，填上画布