请在不影响结果的条件下改变代码的样子：import numpy as np import matplotlib.pyplot as plt x1len = 21 x2len = 18 LEN = x1len + x2len POPULATION_SIZE = 100 GENERATIONS = 251 CROSSOVER_RATE = 0.7 MUTATION_RATE = 0.3 pop = np.random.randint(0,2,size=(POPULATION_SIZE,LEN)) def BinToX(pop): x1 = pop[:,0:x1len] x2 = pop[:,x1len:] x1 = x1.dot(2**np.arange(x1len)[::-1]) x2 = x2.dot(2**np.arange(x2len)[::-1]) x1 = -2.9 + x1*(12 + 2.9)/(np.power(2,x1len)-1) x2 = 4.2 + x2*(5.7 - 4.2)/(np.power(2,x2len)-1) return x1,x2 def func(pop): x1,x2 = BinToX(pop) return 21.5 + x1*np.sin(4*np.pi*x1) + x2*np.sin(20*np.pi*x2) def fn(pop): return func(pop); def selection(pop, fitness): idx = np.random.choice(np.arange(pop.shape[0]), size=POPULATION_SIZE, replace=True, p=fitness/fitness.sum()) return pop[idx] def crossover(IdxP1,pop): if np.random.rand() < CROSSOVER_RATE: C = np.zeros((1,LEN)) IdxP2 = np.random.randint(0, POPULATION_SIZE) pt = np.random.randint(0, LEN) C[0,:pt] = pop[IdxP1,:pt] C[0,pt:] = pop[IdxP2, pt:] np.append(pop, C, axis=0) return def mutation(idx,pop): if np.random.rand() < MUTATION_RATE: mut_index = np.random.randint(0, LEN) pop[idx,mut_index] = 1- pop[idx,mut_index] return best_chrom = np.zeros(LEN) best_score = 0 fig = plt.figure() for generation in range(GENERATIONS): fitness = fn(pop) pop = selection(pop, fitness) if generation%50 == 0: ax = fig.add_subplot(2,3,generation//50 +1, projection='3d', title = "generation："+str(generation)+" best="+str(np.max(fitness))) x1,x2 = BinToX(pop) z = func(pop) ax.scatter(x1,x2,z) for idx in range(POPULATION_SIZE): crossover(idx,pop) mutation(idx,pop) idx = np.argmax(fitness) if best_score < fitness[idx]: best_score = fitness[idx] best_chrom = pop[idx, :] plt.show() print('最优解:', best_chrom, '| best score: %.2f' % best_score)

import numpy as np
import matplotlib.pyplot as plt

x1len = 21
x2len = 18
LEN = x1len + x2len
POPULATION_SIZE = 100
GENERATIONS = 251
CROSSOVER_RATE = 0.7
MUTATION_RATE = 0.3

pop = np.random.randint(0,2,size=(POPULATION_SIZE,LEN))

def BinToX(pop):
x1 = pop[:,0:x1len]
x2 = pop[:,x1len:]
x1 = -2.9 + x1*(12 + 2.9)/(np.power(2,x1len)-1)
x2 = 4.2 + x2*(5.7 - 4.2)/(np.power(2,x2len)-1)
return x1,x2

def func(x1,x2):
return 21.5 + x1*np.sin(4*np.pi*x1) + x2*np.sin(20*np.pi*x2)

def fn(pop):
x1,x2 = BinToX(pop)
return func(x1,x2);

def selection(pop, fitness):
idx = np.random.choice(np.arange(pop.shape[0]),size=POPULATION_SIZE, replace=True, p=fitness/fitness.sum())
return pop[idx]

def crossover(IdxP1,pop):
if np.random.rand() < CROSSOVER_RATE:
C = np.zeros((1,LEN))
IdxP2 = np.random.randint(0, POPULATION_SIZE)
pt = np.random.randint(0, LEN)
C[0,:pt] = pop[IdxP1,:pt]
C[0,pt:] = pop[IdxP2, pt:]
np.append(pop, C, axis=0)
return

def mutation(idx,pop):
if np.random.rand() < MUTATION_RATE:
mut_index = np.random.randint(0, LEN)
pop[idx,mut_index] = 1- pop[idx,mut_index]
return

best_chrom = np.zeros(LEN)
best_score = 0
fig = plt.figure()

for generation in range(GENERATIONS):
fitness = fn(pop)
pop = selection(pop, fitness)

if generation%50 == 0:
ax = fig.add_subplot(2,3,generation//50 +1, projection='3d',title = "generation："+str(generation)+" best="+str(np.max(fitness)))
x1,x2 = BinToX(pop)
z = func(x1,x2)
ax.scatter(x1,x2,z)

for idx in range(POPULATION_SIZE):
crossover(idx,pop)
mutation(idx,pop)

idx = np.argmax(fitness)
if best_score < fitness[idx]:
best_score = fitness[idx]
best_chrom = pop[idx, :]

plt.show()
print('最优解:', best_chrom, '| best score: %.2f' % best_score)