python - 删除 numpy 中的嵌套循环

Question

我一直在编写一个程序来强力检查数字序列以查找欧拉砖，但我想出的方法涉及三重循环。由于嵌套的 Python 循环非常慢，我想知道是否有更好的方法使用 numpy 来创建我需要的值数组。

#x=max side length of brick. User Input.    
for t in range(3,x):
    a=[];b=[];c=[];
    for u in range(2,t):
        for v in range(1,u):
                a.append(t)
                b.append(u)
                c.append(v)
    a=np.array(a)
    b=np.array(b)
    c=np.array(c)
    ...

是否有更好的方法使用 numpy 命令生成数组 af 值？

谢谢。

示例: 如果x=10，当t=3时我想得到:

a=[3]  
b=[2]     
c=[1]

第一次循环。之后，当 t=4 时:

a=[4, 4, 4]
b=[2, 3, 3]
c=[1, 1, 2]

第三次(t=5)我想要:

a=[5, 5, 5, 5, 5, 5]
b=[2, 3, 3, 4, 4, 4]
c=[1, 1, 2, 1, 2, 3]

依此类推，最大边长约为 5000 左右。

编辑:解决方案

a=array(3)
b=array(2)
c=array(1)
for i in range(4,x): #Removing the (3,2,1) check from code does not affect results.
    foo=arange(1,i-1)
    foo2=empty(len(foo))
    foo2.fill(i-1)
    c=hstack((c,foo))
    b=hstack((b,foo2))
    a=empty(len(b))
    a.fill(i)
    ...

现在工作速度快了很多倍。谢谢大家。

Answer 1

尝试使用 .empty 和 .fill (http://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.fill.html)