У меня проблема с функцией conturf matplotlib. У меня есть файл данных txt, из которого я импортирую свои данные. У меня есть столбцы данных (pm1 и pm2), и я выполняю 2D-гистограмму. Я хочу построить эти данные как трехмерную гистограмму и как график контура, чтобы увидеть, где расположены максимальные значения.3D-гистограммы и контурные графики Python
Это мой код:
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
rows = np.arange(200,1300,10)
hist, xedges, yedges = np.histogram2d (pm1_n, pm2_n, bins = (rows, rows))
elements = (len(xedges) - 1) * (len(yedges) - 1)
xpos, ypos = np.meshgrid(xedges[:-1], yedges[:-1])
xpos = xpos.flatten()
ypos = ypos.flatten()
zpos = np.zeros(elements)
dx = 0.1 * np.ones_like(zpos)
dy = dx.copy()
dz = hist.flatten()
#####The problem is here#####
#ax.contourf(xpos,ypos,hist)
#ax.bar3d(xpos, ypos, zpos, dx, dy, dz, zsort='average')
plt.show()
Я могу построить 3d гистограмму, но я не могу построить контур один, если я помещаю hist
в функции contourf я получаю ошибку: Length of x must be number of columns in z
и если I place dz
Я получаю Input z must be a 2D array
Я также пробовал использовать xedges и yexges, но это не решает проблему.
Я думаю, что проблема связана с формой возврата функции histogram2D. Но я не знаю, как это решить.
Я также хотел бы выполнить 3D-график с изменением цветового кода с минимальным до максимального значения. Есть ли способ сделать это?
Спасибо
Ссылки на строки настройки цвета: [pylab example] (http://matplotlib.org/examples/pylab_examples/hist_colormapped.html) и [это сообщение] (http://stackoverflow.com/questions/11950375/apply -Цвет-карта к MPL-инструментарию-mplot3d-axes3d-bar3d) – lanery