29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
|
import numpy as np
|
|||
|
|
import random
|
|||
|
|
|
|||
|
|
def select(individuals, sizepop):
|
|||
|
|
# 本函数对每一代种群中的染色体进行选择，以进行后面的交叉和变异
|
|||
|
|
# individuals input : 种群信息
|
|||
|
|
# sizepop input : 种群规模
|
|||
|
|
# opts input : 选择方法的选择
|
|||
|
|
# ret output : 经过选择后的种群
|
|||
|
|
# yuan=individuals;
|
|||
|
|
individuals['fitness'] = 1 / individuals.get('fitness')
|
|||
|
|
sumfitness = np.sum(individuals.get('fitness'))
|
|||
|
|
sumf = individuals.get('fitness') / sumfitness
|
|||
|
|
index = []
|
|||
|
|
for i in range(sizepop): # 转sizepop次轮盘
|
|||
|
|
pick = random.random()
|
|||
|
|
while pick == 0:
|
|||
|
|
pick = random.random()
|
|||
|
|
for j in range(sizepop):
|
|||
|
|
pick = pick - sumf[j]
|
|||
|
|
if pick < 0:
|
|||
|
|
index.append(j)
|
|||
|
|
break # 寻找落入的区间，此次转轮盘选中了染色体i，注意：在转sizepop次轮盘的过程中，有可能会重复选择某些染色体
|
|||
|
|
# index=unique(index);
|
|||
|
|
# individuals=yuan;
|
|||
|
|
individuals['chrom'] = individuals.get('chrom')[index, :]
|
|||
|
|
individuals['fitness'] = individuals.get('fitness')[index]
|
|||
|
|
ret = individuals
|
|||
|
|
return ret
|