1

假设我有这个 Mathematica 代码，它的输出是一个实数，取决于输入，比如 x,y,z。如何根据代码在 x,y,z 中创建实值函数？

如果代码描述了 x,y,z 之间的简单关系，我可以直接定义这个函数。这里的重点是给定的代码是一个非常复杂的块（或模块）。

例如，如果代码只是简单地将 x,y,z 相加，我会简单地定义

f[x_,y_,z_]=x+y+z

如果我有一个非常复杂的示例，如下所示：

s0[a_, b_, x_] :=
{1, 0, (a + b) x + (1 - a - b)}

s1[a_, b_, c_, d_, p_, q_, n_, x_] :=

Which[0 <= x <= c, {2, n - 1, x/c*q + p},
c <= x <= c + d, {2, n, (x - c)/d*p},
c + d <= x <= 1, {1, n + 1, (x - (c + d))/(1 - c - d)*(1 - a - b)}]

s2[s_, t_, c_, d_, p_, q_, n_, x_] :=

Which[0 <= x <= 1 - s - t, {2, n - 1, 
x/(1 - s - t)*(1 - p - q) + p + q},
1 - s - t <= x <= 1 - s, {3, 
n - 1, (x - (1 - s - t))/t*(1 - c - d) + c + d},
1 - s <= x <= 1, {3, n, (x - (1 - s))/s*d + c}]

s3[c_, a_, b_, s_, t_, n_, x_] :=

Which[0 <= x <= 1 - a - b, {4, n - 1, x/(1 - a - b)*t + 1 - s - t},
1 - a - b <= x <= 1 - a, {4, n, (x - (1 - a - b))/b*(1 - s - t)},
1 - a <= x <= 1, {3, n + 1, (x - (1 - a))/a*c}]

s4[p_, q_, s_, a_, b_, n_, x_] :=

Which[0 <= x <= p, {4, n - 1, x/p*s + 1 - s},
p <= x <= p + q, {5, n - 1, (x - p)/q*a/(a + b) + b/(a + b)},
p + q <= x <= 1, {5, n, (x - (p + q))/(1 - p - q)*b/(a + b)}]

F[{k_, n_, x_}] :=
Which[k == 0, s0[a, b, x],
k == 1, s1[a, b, c, d, p, q, n, x],
k == 2, s2[s, t, c, d, p, q, n, x],
k == 3, s3[c, a, b, s, t, n, x],
k == 4, s4[p, q, s, a, b, n, x]]

G[x_] := NestWhile[F, {0, 0, x}, Function[e, Extract[e, {1}] != 5]]
H[x_] := Extract[G[x], {2}] + Extract[G[x], {3}]
H[0]

要运行上述代码，需要指定列表

{a,b,c,d,p,q,s,t}

并且输出是实数。如何在 a,b,c,d,p,q,s,t 中定义一个函数来输出这些实数？

4

1 回答 1

6

您的基本问题是您的辅助函数中有大量参数，但是您的大字母函数（F和G顺便H说一下 Mathematica 中的单大写字母函数名称是一个坏主意）只需要三个参数，而您的辅助函数（s0等）只返回返回列表中的三个值。

您有两种可能的方法来解决此问题。

您可以重新定义所有内容以要求整个系统所需的所有参数 - 我假设辅助函数中的公共参数名称确实是公共值 - 如下所示：

 G[x_, a_, b_, c_, d_, p_, q_, s_, t_] := 
 NestWhile[F, {0, 0, x, a, b, c, d, p, q, s, t}, 
  Function[e, Extract[e, {1}] != 5]]

或者

您可以设置一些选项，为整个系统全局设置这些参数。向上看Options和OptionsPattern。你会做这样的事情：

首先，定义默认选项：

Options[mySystem] = {aa -> 0.2, bb -> 1., cc -> 2., dd -> 4., 
     pp -> 0.2, qq -> 0.1, ss -> 10., tt -> 20.}
SetOptions[mySystem, {aa->0.2, bb->1., cc->2., dd->4., pp->0.2, 
     qq->0.1, ss->10., tt->20.}]

然后像这样编写你的函数：

F[{k_, n_, x_}, OptionsPattern[mySystem]] :=
 With[{a = OptionValue[aa], b = OptionValue[bb], c = OptionValue[cc], 
   d = OptionValue[dd], p = OptionValue[pp], q = OptionValue[qq], 
   s = OptionValue[ss], t = OptionValue[tt]}, 
  Which[k == 0, s0[a, b, x], k == 1, s1[a, b, c, d, p, q, n, x], 
   k == 2, s2[s, t, c, d, p, q, n, x], k == 3, 
   s3[c, a, b, s, t, n, x], k == 4, s4[p, q, s, a, b, n, x]] ]

您的使用也有一些问题Extract（您假设列表中的部分比前几次迭代中实际存在的部分更多），但这回答了您的主要问题。

于 2012-08-07T06:40:45.210 回答