零基础学习Python--粒子群算法PSO函数寻优-3

Halcom

零基础学习Python--粒子群算法PSO函数寻优-3：

链接：https://pan.baidu.com/s/1yECy3SICh_8a1QZzmtu5ww 提取码：n4ip
将fun.py文件和mian函数写到一起：

1. # -*- coding: utf-8 -*-
   
2. """
   
3. Spyder Editor
   
4. 
   
5. This is a temporary script file.
   
6. """
   
7. 
   
8. from PIL import Image
   
9. from pylab import *
   
10. import matplotlib.pyplot as plt  
    
11. import numpy as np
    
12. from random import *
    
13. #import fun
    
14. 
    
15. def funfitness(x1,x2):
    
16.     y = (x1-0.5)**2+(x2-0.6)**2;
    
17.     return y;
    
18. 
    
19. def minIndex(fitness, fitnesszbest):
    
20.     for i in range(len(fitness)):
    
21.         if(fitness[i] == fitnesszbest):
    
22.             break;
    
23.     return i;
    
24. 
    
25. if __name__=="__main__":
    
26.     c1 = 1.49445;
    
27.     c2 = 1.49445;
    
28.     maxg=200;  
    
29.     sizepop=20;
    
30.     Vmax=1;
    
31.     Vmin=-1;
    
32.     popmax=5;
    
33.     popmin=-5;
    
34.     nVar = 2;
    
35.     #print random.random()
    
36.     # PSO algorithm parameters
    
37.     pop = [[0 for x in range(nVar)] for y in range(sizepop)]
    
38.     gbest = [[0 for x in range(nVar)] for y in range(sizepop)]
    
39.     fitness = [0 for x in range(sizepop)]
    
40.     fitnessgbest = [0 for x in range(sizepop)]
    
41.     zbest = [0 for x in range(nVar)]
    
42.     V = [[0 for x in range(nVar)] for y in range(sizepop)]
    
43.    
    
44.     # init pop
    
45.     for i in range(sizepop):
    
46.         pop[i][0] = popmin + (popmax-popmin)*random();
    
47.         pop[i][1] = popmin + (popmax-popmin)*random();
    
48.         fitness[i] = funfitness(pop[i][0],pop[i][1]);
    
49.         gbest[i][0] = pop[i][0]
    
50.         gbest[i][1] = pop[i][1]
    
51.         fitnessgbest[i] = fitness[i]
    
52.         print(fitness[i])
    
53.     fitnesszbest = min(fitness);
    
54.     index = fun.minIndex(fitness, fitnesszbest);
    
55.     zbest[0] = pop[index][0]
    
56.     zbest[1] = pop[index][1]
    
57.     print(zbest)
    
58.     #plot(range(sizepop), fitness,'r*-')
    
59.     #plot( fitness,'r*-')
    
60.     #show()
    
61.    
    
62.     fitness_iter = [0 for x in range(maxg)]
    
63.     # main loop
    
64.     for i in range(maxg):
    
65.         for j in range(sizepop):
    
66.             # update vel
    
67.             r1 = random();
    
68.             r2 = random();
    
69.             V[j][0] = V[j][0] + c1*r1*(gbest[j][0]-pop[j][0])+c2*r2*(zbest[0]-pop[j][0]);
    
70.             V[j][1] = V[j][1] + c1*r1*(gbest[j][1]-pop[j][1])+c2*r2*(zbest[1]-pop[j][1]);
    
71.             if (V[j][0]>Vmax):
    
72.                 V[j][0] = Vmax;
    
73.             elif (V[j][0]<Vmin):
    
74.                 V[j][0] = Vmin;
    
75.             if (V[j][1]>Vmax):
    
76.                 V[j][1] = Vmax;
    
77.             elif (V[j][1]<Vmin):
    
78.                 V[j][1] = Vmin;
    
79.             # update position
    
80.             pop[j][0] = pop[j][0] + 0.5*V[j][0];
    
81.             pop[j][1] = pop[j][1] + 0.5*V[j][1];
    
82.             if (pop[j][0]>popmax):
    
83.                 pop[j][0] = popmax;
    
84.             elif (pop[j][0]<popmin):
    
85.                 pop[j][0] = popmin;
    
86.             if (pop[j][1]>popmax):
    
87.                 pop[j][1] = popmax;
    
88.             elif (pop[j][1]<popmin):
    
89.                 pop[j][1] = popmin;
    
90.             
    
91.             fitness[j] = funfitness(pop[j][0],pop[j][1]);
    
92.             if (fitness[j]<fitnessgbest[j]):
    
93.                 fitnessgbest[j] = fitness[j]
    
94.                 gbest[j][0] = pop[j][0];
    
95.                 gbest[j][1] = pop[j][1];
    
96.             if (fitness[j]<fitnesszbest):
    
97.                 fitnesszbest= fitness[j];
    
98.                 zbest[0] = pop[j][0];
    
99.                 zbest[1] = pop[j][1];
    
100.         fitness_iter[i] = fitnesszbest;
     
101.    
     
102.     print("最优解:")
     
103.     print(zbest)
     
104.     plot( fitness_iter,'b-' )
     
105.     plt.show()
     

复制代码

therookie

很好 的资源 谢谢楼主