在matplotlib中绘制3d数据

时间:2012-03-31 07:48:02

标签: python matplotlib

- 这是一篇已更新的帖子,旧帖子已删除 -

假设我有类似下面的数据,我想用matplotlib在3D图形/曲面中绘图。我该怎么做?

enter image description here

按照@JouniK.Seppänen的回答,我认为我需要meshgrid(),但我认为我的轴数据不是很正确。如果我有数据(在JSON中),如:

{
    "data": {
        "Random": [
            [1834, 3353, 4602, 5471, 6522, 7873], 
            [2637, 8575, 8357, 10329, 9742, 10359], 
            [3648, 10602, 10667, 10751, 10666, 10552], 
            [4570, 10220, 9202, 10460, 10329, 10928], 
            [5879, 10788, 10832, 10923, 11075, 10989], 
            [6783, 11104, 10235, 10499, 11024, 10731], 
            [7074, 11097, 10222, 10613, 10508, 10767], 
            [7300, 11002, 10727, 11073, 10328, 10864]], 
        "LRU": [
            [123, 155, 201, 223, 296, 321], 
            [143, 590, 1046, 1566, 1924, 2434], 
            [163, 1167, 1774, 2578, 3363, 3980], 
            [182, 1172, 2259, 3038, 4200, 4907], 
            [219, 1718, 3044, 3658, 5236, 5680], 
            [709, 2263, 3588, 4551, 5262, 6197], 
            [2065, 3865, 4430, 5024, 5986, 6617], 
            [3048, 4249, 5029, 5790, 6157, 6826]], 
        "FIFO": [
            [180, 269, 350, 424, 580, 601], 
            [230, 906, 1446, 2009, 2408, 2902], 
            [316, 1590, 2261, 3042, 3848, 4457], 
            [473, 1664, 2781, 3542, 4764, 5398], 
            [665, 2290, 3609, 4194, 5781, 6207], 
            [1158, 2826, 4115, 5064, 5751, 6613], 
            [2457, 4375, 4974, 5471, 6464, 7077], 
            [3512, 4724, 5485, 6272, 6684, 7312]]
    }, 
    "workingSetAxis": [2, 22, 42, 62, 82, 102, 122, 142], 
    "stabilityAxis": [0.0, 0.2, 0.4, 0.6, 0.8, 1.0]
}

其中workingSetAxis应该是x轴,stabilityAxis是y轴。我做了类似

的事情
plot(jsonObj["data"]["FIFO"], jsonObj["workingSetAxis"], jsonObj["stabilityAxis"])

def plot(data, workingSetAxis, stabilityAxis):
    # make axis data
    X, Y = numpy.meshgrid(workingSetAxis, stabilityAxis)
    Z = data

    fig = plt.figure()
    ax = fig.add_subplot(111, projection="3d")
    ax.plot_wireframe(X, Y, Z)
    plt.show()

并得到类似的东西:

enter image description here

与我从excel获得的图表(对于FIFO数据)非常不同

enter image description here

我相信我的轴数据或其他东西是错误的,但是哪个?

2 个答案:

答案 0 :(得分:6)

尝试绘制数据的转置,例如,使用

def plot(data, workingSetAxis, stabilityAxis):
    # make axis data
    X, Y = numpy.meshgrid(workingSetAxis, stabilityAxis)
    Z = numpy.transpose(data) #<--- This is the only line I have changed.

    fig = plt.figure()
    ax = fig.add_subplot(111, projection="3d")
    ax.plot_wireframe(X, Y, Z)
    plt.show()

您的示例数据给出了以下图表: plot demonstrating expected output

答案 1 :(得分:4)

参见例如this example。使用meshgrid将X和Y值转换为2D数组,然后将数据值用作Z。