萤火虫优化算法Python代码

import random
import math

def distance(x, y):
    """计算两个向量之间的欧几里德距离"""
    return math.sqrt(sum([(a - b) ** 2 for a, b in zip(x, y)]))

def brightness(x):
    """计算亮度"""
    return sum(x)

def move(x, v, scale):
    """萤火虫移动"""
    return [a + scale * (b - a) for a, b in zip(x, v)]

def firefly_algorithm(population, distance_func, brightness_func, max_generations=100, alpha=0.5, beta=1.0, gamma=1.0):
    """萤火虫优化算法"""
    # 初始化位置和速度
    swarm = [{'position': x, 'velocity': [0.0] * len(x)} for x in population]
    # 迭代优化
    for generation in range(max_generations):