粒子群优化算法可视化python

粒子群优化算法（Particle Swarm Optimization, PSO）的可视化可以帮助我们更好地理解算法的工作原理和优化过程。下面是一个使用Python实现PSO算法可视化的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 定义目标函数
def target_function(x):
    return x**2

# 定义粒子群优化算法
def particle_swarm_optimization(target_func, num_particles, num_iterations):
    # 初始化粒子群
    particles_position = np.random.uniform(low=-10, high=10, size=num_particles)
    particles_velocity = np.zeros(num_particles)
    particles_best_position = particles_position.copy()
    particles_best_value = np.inf

    # 迭代优化
    global_best_position = np.inf
    global_best_value = np.inf
    for i in range(num_iterations):
        for j in range(num_particles):
            # 更新粒子速度和位置
            particles_velocity[j] = particles_velocity[j] + np.random.uniform(0, 1) * (particles_best_position[j] - particles_position[j]) + np.random.uniform(0, 1) * (global_best_position - particles_position[j])
            particles_position[j] = particles_position[j] + particles_velocity[j]

            # 更新粒子最优位置和最优值
            particle_value = target_func(particles_position[j])
            if particle_value < particles_best_value[j]:
                particles_best_position[j] = particles_position[j]
                particles_best_value[j] = particle_value

            # 更新全局最优位置和最优值
            if particle_value < global_best_value:
                global_best_position = particles_position[j]
                global_best_value = particle_value

        # 可视化粒子位置和目标函数曲线
        x = np.linspace(-10, 10, 100)
        y = target_func(x)
        plt.figure()
        plt.plot(x, y)
        plt.scatter(particles_position, target_func(particles_position), color='red')
        plt.scatter(global_best_position, global_best_value, color='green', marker='x')
        plt.title('Iteration {}'.format(i+1))
        plt.xlabel('x')
        plt.ylabel('y')
        plt.show()

# 调用粒子群优化算法进行可视化
particle_swarm_optimization(target_function, num_particles=30, num_iterations=10)

在该示例中，我们定义了一个简单的目标函数 `target_function`，并实现了粒子群优化算法 `particle_swarm_optimization`。算法中使用了随机初始化的粒子群，并迭代更新粒子的速度和位置。在每次迭代中，我们通过绘制粒子的位置和目标函数曲线来可视化算法的优化过程。

你可以根据需要适当调整粒子数量 `num_particles` 和迭代次数 `num_iterations` 来观察算法的行为。运行代码后，会显示出每次迭代后的目标函数曲线和粒子的位置，并标记出全局最优位置。

注意：这只是一个简单的PSO算法可视化示例，实际应用中，PSO算法可能会根据具体问题进行更复杂的参数设置和适应性调整。