python循环速度提高_提高python中for循环的速度

我正在计算从1到10^7的数字的分区数。我用五边形数定理来实现，它提供了一种计算n的划分数的方法，我们称之为P(n)=P(n-1)+P(n-2)-P(n-5)-P(n-7)+。。。

所以P(n)依赖于不同的P(n-x)，其中x从1到n。总之，长话短说，我使用两个循环，一个从1到10^7，另一个使用x的定义，并根据特定的公式计算。正如您可以想象的，代码执行得相当慢。有什么方法可以提高这些for循环的速度吗？在

另外，我尝试过使用numpy数组，但我对它们并不太满意，不幸的是，它使代码变得更慢。在def build_list(new_limit, k):

list_of_p_minus = list()

while k < new_limit:

list_of_p_minus.append(k*(2*k-1))

list_of_p_minus.append(k*(2*k+1))

k = k+1

return list_of_p_minus

def counter(low, high):

current = low

while current <= high:

yield current

current += 1

def loop_through_numbers(limit, list_of_p_minus):

p_dict = dict()

p_dict[0] = 1

p_dict[1] = 1

aux = 2

aux_index = 1

for number in counter(2, limit+1):

if aux == number:

if aux_index % 4 == 1 or aux_index % 4 == 2:

p_dict[number] = -1

else:

p_dict[number] = 1

aux = aux + (aux_index + 1) * 2

aux_index += 1

else:

p_dict[number] = 0

for element in counter(0, len(list_of_p_minus)):

if element % 4 == 0 or element % 4 == 1:

try:

p_dict[number] += p_dict[number-list_of_p_minus[element]]

except KeyError:

break

else:

try:

p_dict[number] -= p_dict[number-list_of_p_minus[element]]

except KeyError:

break

return

如您所见，我尝试实现迭代器，虽然这使代码运行得更快，但仍然需要大量的时间。在这一点上，任何想法都是非常受欢迎的。。。在