Обновление: Я сделал полную запись того, как я нашел это в своем блоге, http://blog.rtwilson.com/producing-polar-contour-plots-with-matplotlib/ - вы можете сначала проверить его.
Я пытаюсь построить график полярного контура в matplotlib. Я нашел различные ресурсы в Интернете, (а) я не могу заставить код работать, и (б) многие из ресурсов выглядят довольно старыми, и мне интересно, есть ли теперь лучший способ. Например, http://www.mail-archive.com/[email protected]/msg01953.html предполагает, что что-то может быть сделано для улучшения ситуации в ближайшее время, и это было в 2006 году!
Мне бы хотелось, чтобы я мог строить сюжетные сюжетные графики - например, pcolor позволяет делать для своего типа сюжета (см. следующий раздел), но я не могу найти способ сделать это, поэтому Сначала я конвертирую в декартовы координаты.
Во всяком случае, у меня есть следующий код:
from pylab import *
import numpy as np
azimuths = np.arange(0, 360, 10)
zeniths = np.arange(0, 70, 10)
values = []
for azimuth in azimuths:
for zenith in zeniths:
print "%i %i" % (azimuth, zenith)
# Run some sort of model and get some output
# We'll just use rand for this example
values.append(rand())
theta = np.radians(azimuths)
values = np.array(values)
values = values.reshape(len(zeniths), len(azimuths))
# This (from http://old.nabble.com/2D-polar-surface-plot-td28896848.html)
# works fine
##############
# Create a polar axes
# ax = subplot(111, projection='polar')
# pcolor plot onto it
# c = ax.pcolor(theta, zeniths, values)
# show()
r, t = np.meshgrid(zeniths, azimuths)
x = r*np.cos(t)
y = r*np.sin(t)
contour(x, y, values)
Когда я запускаю, я получаю сообщение об ошибке TypeError: Inputs x and y must be 1D or 2D.
. Я не уверен, почему я получаю это, так как x и y - 2D. Я что-то делаю неправильно?
Кроме того, кажется довольно неуклюжим, чтобы вернуть мои значения из моей модели в список и затем изменить его. Есть ли лучший способ сделать это?