基于矩阵的Python 3D绘图

Question

我有一个数据集，是通过……创建的。

import random

count = []
for i in range(1, 4):
    for j in range(3, 6):
        for k in range(15,19):
            count.append((i, j, k, random.random()))

我想画出三个三维图，分别是 count 和 i、j、k 这三组数据的关系（总共三个图）。我查阅了一些例子，地址在这里：http://matplotlib.org/mpl_toolkits/mplot3d/tutorial.html#surface-plots，但是那些例子似乎都是实时生成数据的。我该如何绘制我想要的表面图或网格图呢？

Answer 1

我得赶时间，不过这是我会尝试绘制你数据的方法——就是把 count 和 (i,j) 进行比较的例子——看看最后一行的索引切片。需要的话我很乐意再帮你看看。

>>> from mpl_toolkits.mplot3d import Axes3D
>>> import matplotlib.pyplot as plt
>>> import numpy as np
>>> import random
>>> 
>>> count = []
>>> for i in range(1, 4):
...     for j in range(3, 6):
...         for k in range(15,19):
...             count.append((i, j, k, random.random()))
...             
...         
...     
... 
>>> data = np.array(count)
>>> fig = plt.figure()
>>> ax = fig.gca(projection='3d')
>>> 
>>> # I think this is the sort of thing you want:
>>> 
>>> ax.plot_trisurf(data[:,0], data[:,1], data[:,3])
/usr/local/lib/python2.7/dist-packages/matplotlib/delaunay/triangulate.py:103: DuplicatePointWarni
ng: Input data contains duplicate x,y points; some values are ignored.
  DuplicatePointWarning,
<mpl_toolkits.mplot3d.art3d.Poly3DCollection object at 0x46e8390>
>>> plt.show()