Построение трехмерной поверхности в Julia с помощью Plots или PyPlot

Я хотел бы построить функцию (-ы) с двумя переменными (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")

Если я использую графики таким образом, иногда я получаю пустую трехмерную плоскость без поверхности. Что я делаю неправильно? Я думаю, что это может быть связано с сетками, которые я сделал для kx и ky, но я не уверен.

Изменить: я также получаю следующую ошибку: введите описание изображения здесь


plot julia julia-plots surface
person Madlad    schedule 28.12.2020    source источник

Ответы (1)


arrow_upward
3
arrow_downward

Я изменил кое-что в своем коде.

Сначала я оставил переменные как диапазоны. Во-вторых, я просто вычислил нужные мне функции, не отображая на них переменные. Вот код:

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)")

Я получил то, что хотел! введите описание изображения здесь

person Madlad    schedule 30.12.2020