Python：在 Matplotlib 中绘制均匀分布的球体

Question

我正在尝试创建一个有点像这样的情节：

所有最小值上方都有球体。

表面可以用 sin(x)*sin(y) 图近似：

import numpy as np
import matplotlib.pyplot as plt

def func(x, y):
  return np.sin(2*np.pi*x)*np.sin(2*np.pi*y) / 3

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x = y = np.arange(-1.0, 1.0, 0.05)
X, Y = np.meshgrid(x, y)
zs = np.array([func(x,y) for x,y in zip(np.ravel(X), np.ravel(Y))])
Z = zs.reshape(X.shape)

ax.plot_surface(X, Y, Z, color="grey")
ax.set_zlim3d(-1,1)

plt.show()

但是我不确定如何将均匀分布的球体添加到其中。有人能帮忙吗？

Answer 1

您需要确定函数的最小值（通过您的参数化）在 (x = integer + 0.25, y=integer + 0.75) 或相反。然后您可以简单地使用球坐标对球体进行参数化（例如在此处完成：）并绘制球体。

现在有好消息也有坏消息：

1.) 好消息是正确确定了最小值并创建了球体。在下图中，您可以看到它们位于曲面图蓝色部分的正上方（蓝色部分确实显示了最小值）。

2.) 坏消息是，您将很难找到真正正确渲染球体的另一个角度。我不知道解决这种相当烦人的行为的方法，因此您可能不得不尝试直到找到合适的角度。玩得开心！

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

def func(x, y):
  return np.sin(2*np.pi*x)*np.sin(2*np.pi*y) / 3

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x = y = np.arange(-2.0, 2.0, 0.05)

# Get the minima of the function.
minsx1 = np.arange(int(np.amin(x)) + 0.25, int(np.amax(x)) + 0.25 + 1, 1)
minsy1 = np.arange(int(np.amin(y)) + 0.75, int(np.amax(y)) + 0.75 + 1, 1)
minsx2 = np.arange(int(np.amin(x)) + 0.75, int(np.amax(x)) + 0.75 + 1, 1)
minsy2 = np.arange(int(np.amin(y)) + 0.25, int(np.amax(y)) + 0.25 + 1, 1)

X, Y = np.meshgrid(x, y)
zs = np.array([func(x,y) for x,y in zip(np.ravel(X), np.ravel(Y))])
Z = zs.reshape(X.shape)

# Color map for better detection of minima (blue)
ax.plot_surface(X, Y, Z, cmap="viridis")
ax.set_zlim3d(-1,1)

# Spherical coordinates
r = 0.15
phi = np.linspace(0, 2 * np.pi, 30)
theta = np.linspace(0, np.pi, 30)

# Write spherical coordinates in cartesian coordinates.
x = r * np.outer(np.cos(phi), np.sin(theta))
y = r * np.outer(np.sin(phi), np.sin(theta))
z = r * np.outer(np.ones(np.size(phi)), np.cos(theta))

# Plot the spheres.
for xp in minsx1:
    for yp in minsy1:
        sphere = ax.plot_surface(x+xp, y+yp, z+0.35, color='r')
for xp in minsx2:
    for yp in minsy2:
        sphere = ax.plot_surface(x+xp, y+yp, z+0.35, color='r')
ax.view_init(elev=90, azim=0)
plt.savefig('test.png')
plt.show()

Answer 2

使用matplotlib难免会运行陷入对象被别人隐藏的问题。 matplotlib 3d FAQ and the recommendation is to use mayavi中也有说明。

在 mayavi 中，解决方案如下所示：

from mayavi import mlab
import numpy as np

### SURFACE '''
x,y = np.meshgrid(np.linspace(-2.5,2), np.linspace(-2,2))
f = lambda x,y: .4*np.sin(2*np.pi*x)*np.sin(2*np.pi*y)
z=f(x,y)
mlab.surf(x.T,y.T,z.T, colormap="copper")

### SPHERES '''
px,py = np.meshgrid(np.arange(-2,2)+.25, np.arange(-2,2)+.75)
px,py = px.flatten(),py.flatten()
pz = np.ones_like(px)*0.05
r = np.ones_like(px)*.4
mlab.points3d(px,py,pz,r, color=(0.9,0.05,.3), scale_factor=1)


mlab.show()

Python：在 Matplotlib 中绘制均匀分布的球体

Python: Plotting Evenly Spaced Spheres in Matplotlib

python

3d

plot

matplotlib

mayavi