使用“Plots.jl”将 2d 函数绘制为 3d 空间中的曲面

Posted 2023-02-16

【中文标题】使用“Plots.jl”将 2d 函数绘制为 3d 空间中的曲面

【英文标题】：plotting a 2d function as surface in 3d space with `Plots.jl`

【发布时间】：2017-03-31 17:25:42

【问题描述】：

我在使用Plots.jl 绘图时遇到以下问题。我喜欢绘制 Rosenbrock 函数

rosenbrock(x) = (1.0 - x[1])^2 + 100.0 * (x[2] - x[1]^2)^2

作为表面，期望 2d TupleFloat64,Float64 作为输入。

我能想到的是：

using Plots
gr()

rosenbrock(x) = (1.0 - x[1])^2 + 100.0 * (x[2] - x[1]^2)^2

ts = linspace(-1.0, 1.0, 100)
x = ts
y = map(rosenbrock, [(x, z) for (x,z) in zip(ts,ts)])
z = map(rosenbrock, [(x, y) for (x,y) in zip(ts,ts)])
# plot(x, x, z)
plot(x, y, z, st = [:surface, :contourf])

产生这个情节：

我想我搞砸了一些维度，但我不明白我做错了什么。

我是否必须嵌套 y 和 x 的映射计算才能得到结果？

【问题讨论】：

我建议你试着把你的问题表述得更清楚一点，我不知道你在做什么，我也不知道问题是什么

@isebarn 我已经更新了这个问题。我喜欢将函数绘制为一个曲面，但我不知道该怎么做。

【参考方案1】：

在对我发现的 Rosenbrock 函数进行快速调查后，如果我错了，请纠正我，但您需要指定 y-vector，您不应该将其嵌套在 z 或类似的东西中 其他人尝试了与here 相同的方法，但使用的是 Plots

解决方案如下，由 Patrick Kofod Mogensen

完成

using Plots

function rosenbrock(x::Vector)
  return (1.0 - x[1])^2 + 100.0 * (x[2] - x[1]^2)^2
end

default(size=(600,600), fc=:heat)
x, y = -1.5:0.1:1.5, -1.5:0.1:1.5
z = Surface((x,y)->rosenbrock([x,y]), x, y)
surface(x,y,z, linealpha = 0.3)

这会导致

旁注

我很高兴我搜索了这个，因为我一直在为 Julia 搜索 PyPlot 以外的 3D 绘图仪（因为为我的程序的用户设置可能有点麻烦），而且这看起来更好，而且图片可以旋转。

【讨论】：

非常感谢！ Surface(), surface() 语法让事情变得更容易。

你用了哪些backand和color props？

这看起来（从字体上看）像 PyPlot 后端。