matplotlibのplot_surfaceのプロットに、等高線を2本付け足したい。

Question

pythonのmatplotlibのplt_surfaceで作成したプロットに、
等高線を2本付け足したいと思っています。

2次元の場合は、

fig, ax = plt.subplots(1,1)
fig.colorbar(ax.contourf(X, Y, phasediffY, cmap = "bwr", levels=levels_))
ax.contour(X, Y, phasediffY, levels=[0.3,0.6] )
fig.savefig("hogehoge.png")

とすることで、等高線を2本加えることができました。

3次元の場合のplot_surfaceについて同じような図を作ることはできないでしょうか。高さ0.3と0.6の部分のsurfaceを、線で塗りたいです。plot_surfaceには以下のようなコードを使っています。

fig=plt.figure()
ax=fig.add_subplot(111,projection="3d")
ax.plot_surface(X,Y,phasediffX,cmap="bwr",vmin=0,vmax=1)
fig.savefig("hogehoge.png")

方法の分かる方がいらっしゃったらご教示ください。

環境は次の通りです。

$ python
Python 3.6.5 |Anaconda, Inc.| (default, Apr 29 2018, 16:14:56)
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import matplotlib
>>> matplotlib.__version__
'2.2.2'

Answer 1

surfaceのほうのalpha(透明度)を調整すればいけると思います。

ax.plot_surface(X,Y,phasediffX,cmap="bwr",vmin=0,vmax=1,alpha=0.5)
ax.contour(X, Y, phasediffY, levels=[0.3,0.6])