У меня есть массив 2D numpy, и я хочу построить его в 3D. Я слышал о mplot3d, но не могу нормально работать
Вот пример того, что я хочу сделать. У меня есть массив с размерами (256,1024). Он должен построить трехмерный график, где ось x от 0 до 256, ось y от 0 до 1024, а ось z графика отображает значение массива для каждой записи.
Как мне это сделать?
Похоже, вы пытаетесь создать график surface (альтернативно вы можете нарисовать каркас или заполненный график.
Из информации в вопросе вы можете попробовать что-то вроде:
import numpy
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
# Set up grid and test data
nx, ny = 256, 1024
x = range(nx)
y = range(ny)
data = numpy.random.random((nx, ny))
hf = plt.figure()
ha = hf.add_subplot(111, projection='3d')
X, Y = numpy.meshgrid(x, y) # `plot_surface` expects `x` and `y` data to be 2D
ha.plot_surface(X, Y, data)
plt.show()
Очевидно, вам нужно выбрать более разумные данные, чем использование numpy.random, чтобы получить разумную поверхность.
x, y = range(nx), range(ny) создаст одномерные списки значений x и y, а строка X, Y = numpy.meshgrid(x, y) преобразует эти списки в двумерные массивы NumPy (NumPy должен быть установлен для работы Matplotlib, поэтому он уже есть в вашей системе). Если у вас есть другой вопрос, возможно, вы можете отредактировать свой вопрос, чтобы предоставить больше информации, например, включить данные, которые вы пытаетесь построить.
- person Chris; 10.07.2012
data, а его размер называется nx, ny. Надеюсь, это поможет вам двигаться вперед.
- person Eric O Lebigot; 10.07.2012
from mpl_toolkits.mplot3d import Axes3D в свой сценарий.
- person Chris; 10.07.2012
ha.plot_surface(X.T, Y.T, data) Тогда вы получите правильный результат
- person zwep; 30.11.2018
Вы можете попробовать трехмерный гистограмму, используя функцию bar3d< /а>.
Предположим, у вас есть массив A размерности (25, 10), значение с индексом (i, j) равно A[i][j]. Следующий пример кода может дать вам трехмерный столбчатый график, где высота каждого столбца равна A[i][j].
from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline
np.random.seed(1234)
fig = plt.figure()
ax1 = fig.add_subplot(111, projection='3d')
A = np.random.randint(5, size=(25, 10))
x = np.array([[i] * 10 for i in range(25)]).ravel() # x coordinates of each bar
y = np.array([i for i in range(10)] * 25) # y coordinates of each bar
z = np.zeros(25*10) # z coordinates of each bar
dx = np.ones(25*10) # length along x-axis of each bar
dy = np.ones(25*10) # length along y-axis of each bar
dz = A.ravel() # length along z-axis of each bar (height)
ax1.bar3d(x, y, z, dx, dy, dz)
На моем ПК со случайным начальным числом 1234 я получаю следующий график: 
Однако построение графика для вашей задачи с размерностью (256, 1024) может оказаться медленным.
Вы можете найти ответ в одном из примеров галереи Matplotlib; 3D-примеры ближе к концу.
В более общем плане галерея Matplotlib — отличный первоначальный ресурс для поиска того, как делать некоторые графики.
Примеры, которые я рассмотрел, по существу работают с тремя 2D-массивами: один со всеми значениями x, один со всеми значениями y и последний со всеми значениями z. Таким образом, одним из решений является создание массивов значений x и y (например, с meshgrid()).
Вы также можете использовать модуль oct2py, который на самом деле является мостом python-octave. Используя его, вы можете использовать функции октавы, и вы можете получить то, что вам нужно, и это довольно просто.
ознакомьтесь с этой документацией: https://www.gnu.org/software/octave/doc/v4.0.1/Three_002dDimensional-Plots.html
И для примера примера:
from oct2py import octave as oc
tx = ty = oc.linspace (-8, 8, 41)
[xx, yy] = oc.meshgrid (tx, ty)
r = oc.sqrt (xx * xx + yy * yy) + oc.eps()
tz = oc.sin (r) / r
oc.mesh (tx, ty, tz)
Выше приведен код Python, который аналогичен первому примеру, реализованному в октаве в приведенной выше документации.
