相关疑难解决方法(0)

我一直在谷歌搜索没有成功...也许我的关键字只是糟糕.无论如何,假设我有三个numpy.ndarray相同长度的1D ,我想将它们绘制成3D轨迹.此外,我希望能够做以下任何一件事:

1. 根据函数更改线条的颜色 z
2. 将行的颜色更改为时间的函数(即数组中的索引)

这个演示有一个制作这样一条曲线的例子:

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

fig = plt.figure()
ax = fig.gca(projection='3d')
theta = np.linspace(-4 * np.pi, 4 * np.pi, 100)
z = np.linspace(-2, 2, 100)
r = z**2 + 1
x = r * np.sin(theta)
y = r * np.cos(theta)
ax.plot(x, y, z)

plt.show()

但是我如何实现1或2？欢迎只有一个或另一个的解决方案!提前致谢.