def get_prime_numbers(): primes = [2] for num in range(3, 100): is_prime = True for i in range(2, num): if num % i == 0: is_prime = False break if is_prime: primes.append(num) return primes

使用列表推导式优化后的代码

def get_prime_numbers_list_comprehension(): return [num for num in range(2, 100) if all([num % i for i in range(2, int(num**0.5) + 1)])]

测试

print(get_prime_numbers()) print(get_prime_numbers_list_comprehension())