如何在Julia中查找和替换AST的子表达式

Question

假设我有一个像:(Main.i / (0.5 * Main.i * sin(Main.i)) 。

我想将Main.i每次出现Main.i为其他符号。 朱莉娅有惯用的方法吗？

Answer 1

Main.i不是符号而是表达式，可以通过执行dump(:(Main.i)) 。

以下是我认为可能满足您需求的快速文章：

function expr_replace(expr, old, new)
    expr == old && return new
    if expr isa Expr
        expr = deepcopy(expr) # to avoid mutation of source
        for i in eachindex(expr.args)
            expr.args[i] = expr_replace(expr.args[i], old, new)
        end
    end
    expr
end

对你起作用吗？

编辑：这是一个对deepcopy进行最小安全使用的版本：

function expr_replace(expr, old, new)
    function f(expr)
        expr == old && return deepcopy(new)
        if expr isa Expr
            for i in eachindex(expr.args)
                expr.args[i] = f(expr.args[i])
            end
        end
        expr
    end

    f(deepcopy(expr))
end

总的来说，我认为这无关紧要，因为您可能不想通过此函数传递100行代码。