两个 numpy 数组上的 Python zip

Question

下面是一个工作示例，您可以复制所有变量并运行 for循环检查

我有两个列表，一个长度 = 13，一个长度 = 7。

varying_DA = [70,78,86,90,94,98,100,102,106,110,114,122,130]
test_array = [1,2,3,4,5,6,7,8,9,10,11,12,13] #DUMMY array
varying_Hinv = [60,76,92,100,108,124,140]

我想遍历上面的两个列表，即使用两个 for循环如:

for vary_DA,test in zip(varying_DA,test_array): 

    for vary_Hinv,min_a,max_a in zip(varying_Hinv,min_aH,max_aH):

所以在上面for循环，它以 70 开头对于 vary_DA并循环遍历 vary_Hinv 的 7 个值，即 60,76,...,124,140。然后转到78对于 vary_DA依此类推，直到以 130 结束对于 vary_DA .

我有 min_aH 的特定值和 max_aH对于每个 vary_DA和 vary_Hinv值。

min_aH = np.array([[24,30,30,30,30,30,30],[36,42,42,47,47,47,47],
                   [36,42,42,42,42,42,42],[36,42,42,42,42,42,42],
                   [42,47,47,47,53,47,53],[42,53,47,47,53,53,53],
                   [47,47,47,47,47,53,47],[47,47,47,47,47,59,59],
                   [53,53,59,59,59,59,64],[47,65,59,59,47,47,64],
                   [53,47,41,47,53,47,47],[36,41,41,41,53,47,47],
                   [36,36,41,41,53,47,47]]) 

# As you can see it is a 13x7 matrix, 7 for the 7 different vary_Hinv
# values and 13 for the 13 different vary_DA values    

max_aH = np.array([[54,60,60,60,60,60,60],[60,66,66,66,66,66,66],
                   [66,66,72,72,72,72,72],[66,72,72,72,78,78,78],
                   [72,78,78,78,84,84,84],[78,84,84,89,89,84,84],
                   [78,84,78,89,84,84,78],[84,84,78,84,84,89,89],
                   [84,84,84,89,89,89,89],[84,89,84,89,101,101,101],
                   [89,89,95,101,95,101,101],[96,96,95,107,95,101,101],
                   [96,101,95,107,107,101,101]]) 

即什么时候vary_DA = 70和 vary_Hinv = 60 , min_aH = 24和 max_aH = 54 .什么时候vary_DA = 70和 vary_Hinv = 76 , min_aH = 30和 max_aH = 60等等。

现在我想使用 min_aH 的这些特定值和 max_aH对于 for循环，但正如您从下面的工作示例中看到的那样，它失败了。

ii = 0

for vary_DA,test in zip(varying_DA,test_array): 

    jj = 0 

    for vary_Hinv,min_a,max_a in zip(varying_Hinv,min_aH,max_aH):

        print vary_DA,vary_Hinv,ii,jj,min_aH[ii][jj],max_aH[ii][jj]

        jj = jj + 1

ii = ii + 1

IndexError:标量变量的索引无效。

底线:当我可以清楚地做到以下几点时:

In [12]: min_aH[0]
Out[12]: array([24, 30, 30, 30, 30, 30, 30])

In [13]: min_aH[0][0]
Out[13]: 24

为什么它在 for 中失败环形 ？我如何让它工作？

编辑:

我在上面的工作示例中犯了一个错误。在第二个for循环，print应该是:

print vary_DA,vary_Hinv,ii,jj,min_a[ii][jj],max_a[ii][jj]

我以前有 min_aH和 max_aH而不是 min_a和 max_a .如果以上方法有效，有人可以告诉我吗？

Answer 1

一些评论，首先你的示例使用 python 3 为我运行而没有错误，输出:

70 60 0 0 24 54
70 76 0 1 30 60
70 92 0 2 30 60
...
78 60 0 0 24 54
78 76 0 1 30 60
78 92 0 2 30 60
78 100 0 3 30 60
...

我不相信这是你想要的，因为最后四行的最后两个数字的值应该与前四行不同。我认为您想要显示的最后四行是:

78 60 1 0 36 60
78 76 1 1 42 66
78 92 1 2 42 66
78 100 1 3 47 66

根据您的描述。这是因为 二 = 二 + 1在您的初始循环之外，因此 ii 始终不会更改值。一个工作版本(再次在 python 3.6 中)将是:

ii = 0

for vary_DA,test in zip(varying_DA,test_array): 

    jj = 0 

    for vary_Hinv,min_a,max_a in zip(varying_Hinv,min_aH,max_aH):

        print (vary_DA,vary_Hinv,ii,jj,min_aH[ii][jj],max_aH[ii][jj])

        jj = jj + 1

    ii = ii + 1

虽然这根本没有使用压缩值，如果它试图达到预期的输出将无法实现。你可以在这里看到，如果我打印出压缩值，你只会得到 min_aH 和 max_aH 的每个列表，但我相信你想要单独的值。

ii = 0

for vary_DA,test in zip(varying_DA,test_array): 

    jj = 0 

    for vary_Hinv,min_a,max_a in zip(varying_Hinv,min_aH,max_aH):

        print (vary_DA,vary_Hinv,ii,jj,min_aH[ii][jj],max_aH[ii][jj],min_a,max_a)

        jj = jj + 1

    ii = ii + 1

结果:

70 60 0 0 24 54 [24 30 30 30 30 30 30] [54 60 60 60 60 60 60]
70 76 0 1 30 60 [36 42 42 47 47 47 47] [60 66 66 66 66 66 66]
70 92 0 2 30 60 [36 42 42 42 42 42 42] [66 66 72 72 72 72 72]
...
78 60 1 0 36 60 [24 30 30 30 30 30 30] [54 60 60 60 60 60 60]
78 76 1 1 42 66 [36 42 42 47 47 47 47] [60 66 66 66 66 66 66]
78 92 1 2 42 66 [36 42 42 42 42 42 42] [66 66 72 72 72 72 72]
...

最后，正确使用 zip 来获取您想要的值，从而使迭代器 ii 和 jj 成为不必要的:

ii = 0
for vary_DA,test,min_a,max_a in zip(varying_DA,test_array,min_aH,max_aH): 

    jj = 0 

    for vary_Hinv,mn,mx in zip(varying_Hinv,min_a,max_a):

        print (vary_DA,vary_Hinv,ii,jj,min_aH[ii][jj],max_aH[ii][jj],mn,mx)

        jj = jj + 1
    ii = ii + 1

给予:

70 60 0 0 24 54 24 54
70 76 0 1 30 60 30 60
70 92 0 2 30 60 30 60
...
78 60 1 0 36 60 36 60
78 76 1 1 42 66 42 66
78 92 1 2 42 66 42 66
...

根据需要，最终删除迭代器，如下所示:

for vary_DA,test,min_a,max_a in zip(varying_DA,test_array,min_aH,max_aH): 
    for vary_Hinv,mn,mx in zip(varying_Hinv,min_a,max_a):
        print (vary_DA,vary_Hinv,mn,mx)

这应该会按预期工作。

我还能够使用 python 2.7 成功运行您的示例代码和我的响应，所以我不确定为什么您会根据此处的内容收到该错误。