1

与えられた:

import numpy as np
a = np.arange(6)
b = np.arange(24).reshape(6,4)

私はこのようなものが欲しい：

for i in xrange(len(a)):
    v1 = a[i]
    v2 = b[i,...]

しかし、nditerを使用してそれを行う方法がわかりませんか?

it = np.nditer((a,b))
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-4-7fe57c985cae> in <module>()
----> 1 it = np.nditer((a,b))

ValueError: operands could not be broadcast together with shapes (6) (6,4)

これは単一のオペランドに対して機能しますが、異なるランクのオペランドに対してどのように行うのですか?

a = np.arange(6).reshape(2,3)
for x in np.nditer(a, flags=['external_loop'], order='F'):
...     print x,
4

1 に答える 1

2

なぜ使用しないのzipですか？

>>> for i in xrange(len(a)):
...     print a[i],b[i,...]
... 
0 [0 1 2 3]
1 [4 5 6 7]
2 [ 8  9 10 11]
3 [12 13 14 15]
4 [16 17 18 19]
5 [20 21 22 23]
>>> for v1,v2 in zip(a,b):
...     print v1,v2
... 
0 [0 1 2 3]
1 [4 5 6 7]
2 [ 8  9 10 11]
3 [12 13 14 15]
4 [16 17 18 19]
5 [20 21 22 23]
于 2013-05-23T15:18:52.407 に答える