用Python绘制三维随机游动

Question

我想用Python绘制三维随机游走图。与下图相似的东西。你能给我推荐一个工具吗。我正在尝试使用matplotlib来实现它，但是对于如何实现它感到困惑。

现在，我有一个lattice零数组，它基本上是X*Y*Z多维的，并且保存随机步行者行走的信息，在每个随机步行者所走的(x,y,z)上将0转到1。

如何创建步行的三维视觉？

​

Answer 1

以下是你想要做的事情：

import matplotlib as mpl
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import random

mpl.rcParams['legend.fontsize'] = 10

fig = plt.figure()
ax = fig.gca(projection='3d')

xyz = []
cur = [0, 0, 0]

for _ in xrange(20):
    axis = random.randrange(0, 3)
    cur[axis] += random.choice([-1, 1])
    xyz.append(cur[:])

x, y, z = zip(*xyz)
ax.plot(x, y, z, label='Random walk')
ax.scatter(x[-1], y[-1], z[-1], c='b', marker='o')   # End point
ax.legend()
plt.show()

这会给你一个像这样的随机漫步：

​

​