首页 > 解决方案 > 有条件地在 Julia 的另一个函数中定义一个函数

问题描述

学习较新的 julia 语法和作用域很慢。

在 Julia v1.1.1 中

为什么下面的 MWE 抛出错误“ff 未定义”的解释是什么?

N = 5;
typp = "b";
mu = 2;

function bigfun()

  function f(u,mu)
    ee = mu*u;
    return ee
  end

  function g(uv,mu)
    ee = (mu^2)*uv
    return ee;
  end

  while 1 == 1

    u = ones(N);
    if typp == "a"
      ff(u) = f(u,mu);
    elseif typp == "b"
      ff(u) = g(u,mu);
    end
    fu = ff(u);
    break;

  end

end

bigfun();

标签: julia

解决方案


这是 Julia 中的一个已知错误:https ://github.com/JuliaLang/julia/issues/15602 。您不能有条件地定义内部函数。有几种方法可以解决这个问题:

  1. 定义ff为匿名函数:
        if typp == "a"
          ff = u -> f(u,mu)
        elseif typp == "b"
          ff = u -> g(u,mu)
        end
        fu = ff(u)
    
  2. 定义ff一次,并在其中添加条件:
        function ff(u, typp)
          if typp == "a"
            f(u,mu)
          elseif typp == "b"
            g(u,mu)
          end
        end
        fu = ff(u, typp)
    
  3. 根本不定义函数ff。您不需要,在您提供的示例中,只需fu有条件 地分配
        if typp == "a"
          fu = f(u,mu)
        elseif typp == "b"
          fu = g(u,mu)
        end
    

推荐阅读