mathematica 分段函数使用不同的颜色

最新推荐文章于 2024-06-09 16:52:17 发布

ppdouble

最新推荐文章于 2024-06-09 16:52:17 发布

阅读量1.8k

点赞数

分类专栏：数学文章标签： 1024程序员节

本文链接：https://blog.csdn.net/ppdouble/article/details/120930648

版权

数学专栏收录该内容

12 篇文章

订阅专栏

环境: mathematica 12.0, fedora 29 workstation x86_64
smooth L1:
$\begin{cases} 0.5x^2 & \mid x\mid<1\\ \mid x\mid-0.5 & \text{otherwise} \end{cases}$

code snipet 1:

f = Piecewise[{{0.5 #^2, Abs[#] < 1}, {Abs[#] - 0.5,  Abs[#] >= 1}, {Log[#], 2 < #}}] &;

colorFunction = f;
piecewiseParts = Length@colorFunction[[1, 1]];
colors = ColorData[1][#] & /@ Range@piecewiseParts;
colorFunction[[1, 1, All, 1]] = colors;

Plot[f[x], {x, -2, 2}, ColorFunction -> colorFunction, ColorFunctionScaling -> False]

code snipet 2:

pwSplit[_[pairs : {{_, _} ..}]] := 
 Piecewise[{#}, Indeterminate] & /@ pairs

pwSplit[_[pairs : {{_, _} ..}, expr_]] := 
 Append[pwSplit@{pairs}, pwSplit@{{{expr, Nor @@ pairs[[All, 2]]}}}]

pw = Piecewise[{{0.5 x^2, Abs[x] < 1}, {Abs[x] - 0.5, Abs[x] >= 1}}];

Plot[Evaluate[pwSplit@pw], {x, -2, 2}, PlotStyle -> Thick,  Axes -> True]

smoothL1