matplotlib 在 3D 中绘制多条线

Question

我正在尝试使用matplotlib在3D绘图中绘制多条线。我有 6 个具有 x 和 y 值的数据集。到目前为止，我尝试的是，给数据集中的每个点一个 z 值。因此，数据集 1 中的所有点都有 z=1，数据集 2 的所有点都有 z=2，依此类推。然后我将它们导出到三个文件中。"X.txt"包含所有 x 值，"Y.txt"包含所有 y 值，"Z.txt"也是如此。

这是到目前为止的代码：

        #!/usr/bin/python
        from mpl_toolkits.mplot3d import axes3d
        import matplotlib.pyplot as plt
        import numpy as np
        import pylab
        xdata = '/X.txt'
        ydata = '/Y.txt'
        zdata = '/Z.txt'
        X = np.loadtxt(xdata)
        Y = np.loadtxt(ydata)
        Z = np.loadtxt(zdata)
        fig = plt.figure()
        ax = fig.add_subplot(111, projection='3d')
        ax.plot_wireframe(X,Y,Z)
        plt.show()

我得到的看起来非常接近我需要的。但是，使用线框时，每个数据集的第一个点和最后一个点是连接的。如何更改每个数据集的线的颜色以及如何删除数据集之间的连接线？

有没有比线框更好的绘图风格？

Answer 1

单独加载数据集，然后单独绘制每个数据集。

我不知道你有什么格式，但你想要这样的东西

from mpl_toolkits.mplot3d.axes3d import Axes3D
import matplotlib.pyplot as plt
fig, ax = plt.subplots(subplot_kw={'projection': '3d'})
datasets = [{"x":[1,2,3], "y":[1,4,9], "z":[0,0,0], "colour": "red"} for _ in range(6)]
for dataset in datasets:
    ax.plot(dataset["x"], dataset["y"], dataset["z"], color=dataset["colour"])
plt.show()

每次您在axes对象上调用plot（或plot_wireframe但我不知道您需要什么）时，它都会将数据添加为新系列。如果你省略color参数，matplotlib会为你选择它们，但它不是太聪明，在你添加太多系列后，它会循环并再次开始使用相同的颜色。

注意：我还没有测试过这个 - 不记得color是否正确的论点。不过很确定。