ackley 函数python 代码

以下是ackley函数的Python代码：

import math
import numpy as np

def ackley(x):
    n = 2
    a = 20
    b = 0.2
    c = 2 * np.pi
    s1 = 0
    s2 = 0
    for i in range(0, n):
        s1 = s1 + x[i]**2
        s2 = s2 + math.cos(c * x[i])
    y = -a * math.exp(-b * math.sqrt(1/n * s1)) - math.exp(1/n * s2) + a + math.exp(1)
    return y

其中，`x`是一个长度为2的一维数组，表示函数的自变量。函数返回值为函数在给定自变量下的函数值。