Julia 函数定义：function 与箭头函数

在 Julia 中定义函数有两种主要方式：使用 function 关键字和使用箭头（->）语法。它们功能相似，但在可读性、适用场景和语法限制上有明显区别。掌握两者的差异能让你写出更简洁或更清晰的代码。

使用 function 关键字定义函数

输入以下代码来定义一个标准函数：

function add(x, y)
    return x + y
end

这种写法结构清晰，适合多行逻辑或复杂计算。即使只有一行，也可以省略 return，Julia 会自动返回最后一行的值：

function square(x)
    x * x
end

你还可以将函数写在一行内（仍用 function）：

function greet(name) "Hello, $name!" end
```

但这种写法较少见，通常用于极简函数。

---

### 使用箭头函数（匿名函数）

**创建**一个箭头函数的语法是 `(参数) -> 表达式`。例如：

```julia
add = (x, y) -> x + y
```

这等价于上面的 `function add(x, y)`，但它是匿名的——没有函数名，只是把函数赋值给变量 `add`。

箭头函数**必须**是单表达式，不能包含多行语句或 `return`。它主要用于：

- 作为高阶函数的参数（如 `map`, `filter`）
- 临时、一次性的简单操作

**调用** `map` 时使用箭头函数：

```julia
numbers = [1, 2, 3, 4]
squares = map(x -> x^2, numbers)
```

这里 `x -> x^2` 是一个匿名函数，对每个元素平方。

---

### 关键区别对比

| 特性 | `function` 定义 | 箭头函数 |
| :--- | :---: | :---: |
| 是否有名字 | 有（如 `add`） | 无（匿名，需赋值给变量才有引用） |
| 支持多行吗 | **支持** | ❌ 仅单表达式 |
| 能用 `return` 吗 | **能** | ❌ 自动返回表达式结果 |
| 适合复杂逻辑吗 | **适合** | 不适合 |
| 常用于高阶函数吗 | 较少（除非提前定义） | **常用** |

---

### 何时用哪种？

**优先使用 `function` 的情况**：
- 函数逻辑超过一行
- 需要添加类型注解（如 `function f(x::Int)`）
- 函数会被多次调用或需要文档说明
- 你想让代码更容易调试（有明确函数名）

**优先使用箭头函数的情况**：
- 在 `map`, `filter`, `reduce` 等函数中做简单变换
- 写测试或脚本时快速定义临时逻辑
- 函数体只有一个表达式且无需复用

例如，筛选偶数：

```julia
evens = filter(x -> x % 2 == 0, 1:10)
```

比先定义一个 `is_even(x)` 函数更直接。

---

### 注意作用域和性能

两种方式在性能上**没有区别**。Julia 编译器会将它们优化成相同的底层代码。

但要注意：箭头函数中的变量捕获遵循闭包规则。例如：

```julia
a = 10
f = x -> x + a
```

此时 `f` 捕获了外部的 `a`。如果后续修改 `a`，**不会影响**已创建的 `f`，因为 Julia 在创建闭包时会绑定当前值（对不可变类型而言）。但对于可变对象（如数组），行为可能不同，需谨慎。

---

### 类型注解的写法差异

给 `function` 添加类型很简单：

```julia
function divide(x::Float64, y::Float64)::Float64
    x / y
end
```

而箭头函数**不能直接写类型注解**。如果你需要类型约束，必须用 `function`，或者在外层包装：

```julia
safe_divide = (x, y) -> begin
    @assert y != 0 "Division by zero"
    x / y
end
```

但即便如此，也无法在箭头函数签名中声明类型。因此，涉及类型安全或多重派发时，**必须使用 `function`**。

---

### 多重派发只适用于 `function`

Julia 的核心特性之一是多重派发（multiple dispatch）：同一个函数名根据参数类型执行不同逻辑。

**定义**多个同名函数：

```julia
function process(x::Int)
    "Integer: $x"
end

function process(x::String)
    "String: $x"
end

这时 process(5) 和 process("hi") 会调用不同版本。

但箭头函数无法参与多重派发，因为它没有函数名，只是匿名函数对象。你不能对同一个变量名多次赋值不同类型的箭头函数来实现派发。

因此，只要涉及多重派发，必须使用 function。

实际示例：两种写法转换

假设你想写一个函数计算向量的欧氏范数。

用 function：

function norm2(v)
    sqrt(sum(x -> x^2, v))
end

这里外层用 function，内层 sum 的参数用了箭头函数——这是典型混合用法。

若强行全用箭头函数：

norm2 = v -> sqrt(sum(x -> x^2, v))

两者效果相同。但前者更易扩展（比如加错误检查），后者更紧凑。

常见错误

错误1：在箭头函数里写多行

f = x -> begin
    y = x + 1
    y * 2
end

虽然 begin...end 在技术上是一个表达式，但这种写法违背箭头函数“简洁”的初衷，应改用 function。

错误2：试图给箭头函数加方法

f = x -> x + 1
f(x::String) = x * "!"

这会报错，因为 f 是一个变量（绑定到函数对象），不是泛型函数名。只有 function f(...) 定义的 f 才能添加新方法。

总结选择策略

1. 写工具函数、库函数、带文档的函数 → 用 function
2. 在 map/filter/sort 等中做简单变换 → 用箭头函数
3. 需要类型注解或多分派 → 必须用 function
4. 逻辑超过一行或需调试 → 用 function

记住：箭头函数是语法糖，不是替代品。合理混用，才能发挥 Julia 的表达力。