5

我正在尝试使用 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()

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

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

4

1 回答 1

8

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

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

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()

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

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

于 2013-10-30T11:19:03.580 回答