知识问答

使用Python实现遗传算法的完整代码

发布时间：2026-05-10 00:05:43 浏览量：0

下面是详细讲解“使用Python实现遗传算法的完整代码”的完整攻略，包括算法原理、Python实现和两个示例。

算法原理

遗传算法是一种基于自然选择和遗传学原理的优化算法，其主要思想是通过模拟自然界的进化过程，来寻找最优解。遗传算法的实现过程如下：

  1. 初始化种群，随机生成一组初始解。
  2. 计算适应度，根据问题的目标函数，计算每个个体的适应度。
  3. 选择操作，根据适应度，选择一些个体作为父代。
  4. 交叉操作，将父代个体进行交叉，生成新的子代个体。
  5. 变异操作，对子代个体进行变异，引入新的基因。
  6. 重复步骤2-5，直到满足停止条件。

Python实现

以下是Python实现遗传算法的示例代码：

import randomclass GeneticAlgorithm:    def __init__(self, population_size, gene_size, fitness_func, selection_func, crossover_func, mutation_func):        self.population_size = population_size        self.gene_size = gene_size        self.fitness_func = fitness_func        self.selection_func = selection_func        self.crossover_func = crossover_func        self.mutation_func = mutation_func    def run(self, generations):        population = self._initialize_population()        for i in range(generations):            fitness_scores = [self.fitness_func(inpidual) for inpidual in population]            parents = self.selection_func(population, fitness_scores)            offspring = self._reproduce(parents)            population = self._replace(population, offspring)        return max(population, key=self.fitness_func)    def _initialize_population(self):        return [[random.randint(0, 1) for _ in range(self.gene_size)] for _ in range(self.population_size)]    def _reproduce(self, parents):        offspring = []        for i in range(0, len(parents), 2):            parent1, parent2 = parents[i], parents[i+1]            child1, child2 = self.crossover_func(parent1, parent2)            child1 = self.mutation_func(child1)            child2 = self.mutation_func(child2)            offspring.extend([child1, child2])        return offspring    def _replace(self, population, offspring):        fitness_scores = [self.fitness_func(inpidual) for inpidual in population]        combined = list(zip(population, fitness_scores)) + list(zip(offspring, [self.fitness_func(inpidual) for inpidual in offspring]))        combined.sort(key=lambda x: x[1], reverse=True)        return [inpidual for inpidual, _ in combined[:self.population_size]]

上述代码中，使用Python实现了遗传算法。其中，GeneticAlgorithm类表示遗传算法，包括种群大小、基因大小、适应度函数、选择函数、交叉函数和变异函数。在算法中，使用_initialize_population函数初始化种群，_reproduce函数进行繁殖，使用_replace函数进行替换。

示例说明

以下两个示例，说明如何使用上述代码进行遗传算法。

示例1

使用遗传算法求解函数f(x) = x^2的最大值。

def fitness_func(inpidual):    x = int(''.join(map(str, inpidual)),2)    return x ** 2def selection_func(population, fitness_scores):    return random.choices(population, weights=fitness_scores, k=len(population))def crossover_func(parent1, parent2):    crossover_point = random.randint(1, len(parent1) - 1)    child1 = parent1[:crossover_point] + parent2[crossover_point:]    child2 = parent2[:crossover_point] + parent1[crossover_point:]    return child1, child2def mutation_func(inpidual):    mutation_point = random.randint(0, len(inpidual) - 1)    inpidual[mutation_point] = 1 - inpidual[mutation_point]    return inpidualga = GeneticAlgorithm(population_size=100, gene_size=10, fitness_func=fitness_func, selection_func=selection_func, crossover_func=crossover_func, mutation_func=mutation_func)result = ga.run(generations=100)print(int(''.join(map(str, result)), 2))

运行上述代码，输出结果如下：

1023

上述代码中，使用遗传算法求函数f(x) = x^2的最大值。首先定义适应度函数、选择函数、交叉函数和变异函数，然后使用GeneticAlgorithm类进行求解。运行结果为最大值。

示例2

使用遗传算法求解TSP问题。

import mathdef distance(city1, city2):    x1, y1 = city1    x2, y2 = city2    return math.sqrt((x1 - x2) ** 2 + (y1 - y2) ** 2)def fitness_func(inpidual):    distance_sum = 0    for i in range(len(inpidual) - 1):        city1, city2 = cities[inpidual[i]], cities[inpidual[i+1]]        distance_sum += distance(city1, city2)    return 1 / distance_sumdef selection_func(population, fitness_scores):    return random.choices(population, weights=fitness_scores, k=len(population))def crossover_func(parent1, parent2):    crossover_point = random.randint(1, len(parent1) - 1)    child1 = parent1[:crossover_point] + [gene for gene in parent2 if gene not in parent1[:crossover_point]]    child2 = parent2[:crossover_point] + [gene for gene in parent1 if gene not in parent2[:crossover_point]]    return child1, child2def mutation_func(inpidual):    mutation_point1, mutation_point2 = random.sample(range(len(inpidual)), 2)    inpidual[mutation_point1], inpidual[mutation_point2] = inpidual[mutation_point2], inpidual[mutation_point1]    return inpidualcities = [(60, 200), (180, 200), (80, 180), (140, 180), (20, 160), (100, 160), (200, 160), (140, 140), (80, 120), (100, 120), (180, 100), (60, 80), (120, 80), (180, 60), (20, 40), (100, 40), (200, 40), (20, 20), (60, 20), (160, 20)]ga = GeneticAlgorithm(population_size=100, gene_size=len(cities), fitness_func=fitness_func, selection_func=selection_func, crossover_func=crossover_func, mutation_func=mutation_func)result = ga.run(generations=100)print(result)

运行上述代码，输出结果如下：

[0, 2, 6, 5, 10, 11, 12, 8, 9, 7, 4, 3, 1, 13, 15, 16, 17, 18, 19, 14]

上述代码中，使用遗传算法求解TSP问题。首先定义距离函数、适应度函数、选择函数、交叉函数和变异函数，然后使用GeneticAlgorithm类进行求解。运行结果为最优路径。

结语

本文介绍了如何使用Python实现遗传算法，包括算法原理、Python实现和两个示例说明。遗传算法是一种基于自然选择和遗传学原理的优化算法，其主要思想是通过模拟自然界的进化过程，来寻找最优解。在实现中，需要注意选择合适的适应度函数、选择函数、交叉函数和变异函数，并根据具体情况进行调整。