Matplotlib v0.99 Поверхность с контуром: zdir не работает

Question

Я пытаюсь построить функцию f (x, y) = (x + 2) * y ^ 2 с некоторыми кривыми изо уровня, спроецированными на плоскость x-y Код, который я использовал, это:

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FixedLocator, FormatStrFormatter
import matplotlib.pyplot as plt
import numpy as np
from mpl_toolkits.mplot3d import axes3d, Axes3D
import pylab as p


fig = plt.figure()
ax = Axes3D(fig)
X = np.arange(0, 2.5, 0.1)
Y = np.arange(0, 2.5, 0.1)
X, Y = np.meshgrid(X, Y)

Z = ((X+2))*(Y**2)
surf = ax.plot_surface(X, Y, Z,rstride=1, cstride=1, alpha=0.3, cmap=cm.jet)
cset=plt.contour(X,Y,Z,zdir='z',offset=0)


ax.clabel(cset, fontsize=9, inline=1)
ax.set_zlim3d(0, 30)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()

Моя проблема в том, что zdir не работает, то есть контурные линии находятся на поверхности, а не на плоскости x-y.

Есть идеи? Заранее спасибо

Answer 1

Ваш код отлично работает для меня (matplotlib 1.0.1).

Кстати: похоже, этот пример был добавлен в галерею с matplotlib 1.0.Может быть, это проблема предыдущей версии?

Answer 2

zdir определяет направление к проекту. (zdir='x' проецируется вдоль оси x) offset определяет местоположение плоскости, на которую будет проецироваться (вдоль оси, определяемой zdir)

Пример

Я собираюсь угадать, что вы хотите:

cset=plt.contour(X,Y,Z,zdir='z',offset=30)

Answer 3

-> xs и ys: это координаты для осей x и y -> zs: это значение (я) для оси z. Это может быть один для всех точек или один для каждой точки -> zdir: это выберет размер оси z (обычно это zs, но может будь хз или уй)