任意嵌套for循环数的算法

2024-03-29 12:24:08 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图找出一种算法,将我的代码概括为一个参数的任意整数值,该参数在我的代码中控制嵌套for循环的数量。我的代码看起来像

A = numpy matrix of dimension (n x m)

For i = 0:p
 For j = 0:q
  ...
   For l = 0:z

     X[counter] = A[0]*i + A[1]*j + ... + A[n]*l
     counter = counter + 1

X = numpy matrix of dimension (p*q*...*z x len(A[0]))

其中for循环的数量由任意整数控制。你知道吗

谢谢你的建议!你知道吗


Tags: of代码numpy算法for参数数量len
2条回答

正如所指出的,这将很快爆发。但是,您可以使用笛卡尔积迭代器:

import itertools
import numpy as np

# Create a list with ranges:
rngs=np.array([p,q,..,z])

#Initialize X empty
X = np.empty((rngs.prod(), A.shape[0]))

#Cycle over all cartesian products
for i,t in enumerate(itertools.product(*[range(i) for i in rngs])):
    X[i,:] = sum([A[i,:] * t[i] for i in range(len(t))])

试验数据:

A = np.random.randint(10, size=(10, 10))
A
array([[8, 5, 0, 2, 0, 4, 5, 5, 0, 9],
       [7, 0, 5, 9, 9, 4, 8, 2, 6, 8],
       [4, 3, 8, 5, 2, 5, 4, 8, 6, 1],
       [0, 5, 6, 5, 5, 0, 8, 5, 4, 9],
       [3, 3, 2, 6, 6, 9, 7, 7, 3, 3],
       [4, 0, 7, 2, 3, 2, 2, 4, 1, 2],
       [6, 2, 5, 9, 9, 9, 4, 7, 7, 3],
       [6, 3, 4, 9, 0, 3, 8, 1, 6, 8],
       [6, 5, 6, 0, 8, 7, 9, 0, 7, 4],
       [2, 1, 4, 1, 3, 8, 3, 3, 2, 9]])

rngs = np.random.randint(1, 5, size=10)
rngs
array([4, 2, 1, 2, 2, 3, 4, 4, 2, 4])

X = np.empty((rngs.prod(), A.shape[0]))

for i,t in enumerate(itertools.product(*[range(i) for i in rngs])):
    X[i,:] = sum([A[i,:] * t[i] for i in range(len(t))])

X
array([[  0.,   0.,   0., ...,   0.,   0.,   0.],
       [  2.,   1.,   4., ...,   3.,   2.,   9.],
       [  4.,   2.,   8., ...,   6.,   4.,  18.],
       ...,
       [ 86.,  44.,  64., ...,  64.,  63.,  97.],
       [ 88.,  45.,  68., ...,  67.,  65., 106.],
       [ 90.,  46.,  72., ...,  70.,  67., 115.]])

首先:你确定要这样做吗?实际执行的语句数量会随着循环数量的增加而迅速增加,除了极少量的循环之外,您的代码在任何情况下都会变得很慢,可能会慢到无法使用。我会考虑你是否真的需要这些环。你知道吗

撇开这一点不谈,我会递归地这样做:

def nextloop(counter, coefficients, vectors):
    counter = counter - 1
    if counter == 0:
        result = 0
        for i in range(len(vectors)):
            result += coefficients[i] * vectors[i]
        return result
    else:
        return nextloop(counter, coefficients, vectors)

finalresult = nextloop(5, coeffs, vecs) # Adjust initial call appropriately

这是不完全正确的,因为我不太清楚你的算法在每个循环上做什么,但你可以根据需要修改它。你知道吗

相关问题 更多 >