在Julia中绘制3D曲面，使用Plots或PyPlot。

Question

我想绘制一个包含两个变量（在代码中为e_pos和e_neg）的函数图像。其中，t和a是常数，我已经给它们赋值为1。

绘制这个函数的代码如下：

t = 1
a = 1

kx = ky = range(3.14/a, step=0.1, 3.14/a)

# Doing a meshgrid for values of k
KX, KY = kx'.*ones(size(kx)[1]), ky'.*ones(size(ky)[1])


e_pos = +t.*sqrt.((3 .+ (4).*cos.((3)*KX*a/2).*cos.(sqrt(3).*KY.*a/2) .+ (2).*cos.(sqrt(3).*KY.*a)));
e_neg = -t.*sqrt.((3 .+ (4).*cos.((3)*KX*a/2).*cos.(sqrt(3).*KY.*a/2) .+ (2).*cos.(sqrt(3).*KY.*a)));

using Plots


plot(KX,KY,e_pos, st=:surface,cmap="inferno")

如果我这样使用Plots，有时会得到一个空的3D平面而没有表面。我做错了什么？我认为这可能与我为kx和ky做的网格有关，但我不确定。

编辑：我还遇到了以下错误：

- Madlad

Answer 1

我在代码中做了一些修改。首先，我将变量保留为范围。其次，我直接计算所需的函数，而不是将变量映射到它们上面。以下是代码：

t = 2.8
a = 1

kx = range(-pi/a,stop = pi/a, length=100)
ky = range(-pi/a,stop = pi/a, length=100)

#e_pos = +t*np.sqrt(3 + 4*np.cos(3*KX*a/2)*np.cos(np.sqrt(3)*KY*a/2) + 2*np.cos(np.sqrt(3)*KY*a))

e_pos(kx,ky) = t*sqrt(3+4cos(3*kx*a/2)*cos(sqrt(3)*ky*a/2) + 2*cos(sqrt(3)*ky*a))
e_neg(kx,ky) = -t*sqrt(3+4cos(3*kx*a/2)*cos(sqrt(3)*ky*a/2) + 2*cos(sqrt(3)*ky*a))

# Sort of broadcasting?
e_posfunc = e_pos.(kx,ky);
e_negfunc = e_neg.(kx,ky);

对于绘图，我仅使用了GR后端：

using Plots
gr()

plot(kx,ky,e_pos,st=:surface)
plot!(kx,ky,e_neg,st=:surface, xlabel="kx", ylabel="ky",zlabel="E(k)")

我得到了我想要的东西！

- Madlad