Список итераций трехмерного массива всех элементов с помощью nditer () из numpy 1.15.3 - PullRequest
Новая
       10

Список итераций трехмерного массива всех элементов с помощью nditer () из numpy 1.15.3

0 голосов
/ 31 октября 2018

У меня есть список трехмерных массивов Numpy, и я хочу перебрать каждый элемент этой структуры и внести некоторые изменения с помощью if ifmets. Код ниже выполняет то, что я хочу сделать: 

for counter1, entry in enumerate(all_frames_flow):
    for counter2, entry2 in enumerate(entry):
        for counter3, entry3 in enumerate(entry2):
            for counter4, entry4 in enumerate(entry3):
                if entry4 < -20.0:
                    all_frames_flow[counter1][counter2][counter3][counter4]=-20.0
                if entry4 > 20.0:
                    all_frames_flow[counter1][counter2][counter3][counter4]=20.0
                all_frames_flow[counter1][counter2][counter3][counter4]/=20

Но мне было интересно, есть ли более питонический путь. В numpy >=1.15.0 я попробовал этот новый код из документации, но он не работает, он не возвращает желаемых результатов, я вижу значения больше abs(20), и мне интересно, почему это так: 

for counteref, _ in enumerate(backup2):                    
    with np.nditer(backup2[counteref], op_flags=['readwrite'], order = 'K') as it:
            for x in it:
                #print x
                if (x < -20.0):
                    x=-20.0
                if (x > 20.0):
                    x = 20.0
                x/=20.0

Ответы [ 2 ]

0 голосов
/ 31 октября 2018

Прежде чем пытаться использовать лучшие / альтернативные итераторы, вы должны попытаться выполнить задачу без итерации (то есть сделать что-то с скомпилированным numpy кодом) 

In [347]: arr = np.random.randint(-40,40,(2,3,4))

Например, есть метод clip: 

In [348]: arr.clip(-20, 20)
Out[348]: 
array([[[-20, -20,  20,  -6],
        [-15, -17,  -8, -20],
        [  2, -20, -16,  20]],

       [[-20,   3, -20,  17],
        [ 20,  20,  20, -17],
        [ 11, -20,  20,   0]]])

и деление всего на 20 тривиально в numpy: 

In [349]: _/20
Out[349]: 
array([[[-1.  , -1.  ,  1.  , -0.3 ],
        [-0.75, -0.85, -0.4 , -1.  ],
        [ 0.1 , -1.  , -0.8 ,  1.  ]],

       [[-1.  ,  0.15, -1.  ,  0.85],
        [ 1.  ,  1.  ,  1.  , -0.85],
        [ 0.55, -1.  ,  1.  ,  0.  ]]])

Еще лучше, научитесь делать подобные вещи с логическим маскированием: 

In [351]: arr
Out[351]: 
array([[[-32, -30,  39,  -6],
        [-15, -17,  -8, -34],
        [  2, -31, -16,  35]],

       [[-39,   3, -37,  17],
        [ 31,  30,  28, -17],
        [ 11, -24,  26,   0]]])

In [354]: mask1 = arr<-20
In [355]: mask2 = arr>20
In [356]: mask1
Out[356]: 
array([[[ True,  True, False, False],
        [False, False, False,  True],
        [False,  True, False, False]],

       [[ True, False,  True, False],
        [False, False, False, False],
        [False,  True, False, False]]])
In [357]: mask2
Out[357]: 
array([[[False, False,  True, False],
        [False, False, False, False],
        [False, False, False,  True]],

       [[False, False, False, False],
        [ True,  True,  True, False],
        [False, False,  True, False]]])
In [358]: arr[mask1]=-20
In [359]: arr[mask2]=20
In [360]: arr
Out[360]: 
array([[[-20, -20,  20,  -6],
        [-15, -17,  -8, -20],
        [  2, -20, -16,  20]],

       [[-20,   3, -20,  17],
        [ 20,  20,  20, -17],
        [ 11, -20,  20,   0]]])

Что касается вашей итерации, важно помнить, что в любой итерации Python вы не можете использовать

для х в ...: х = -20,0

чтобы изменить источник. Это присваивание x=... присваивает новое значение переменной x и разрывает ее связь с итерацией. Попробуйте это с простым списком, если вы не понимаете, почему. Вы должны изменить переменную на месте. Если x - простое целое число, которое невозможно.

На первой итерации вы индексируете и изменяете all_frames_flow, так что это работает:

all_frames_flow [counter1] [counter2, counter3, counter4] = - 20,0

nditer предоставляет изменяемую переменную итерации, поэтому вы можете сделать: 

In [364]: with np.nditer(arr, op_flags=['readwrite'], order = 'K') as it:
     ...:    for x in it:
     ...:        #print x
     ...:        if (x < -20.0):
     ...:            x[...]=-20.0      # change x in-place
     ...:        if (x > 20.0):
     ...:            x[...] = 20.0
     ...:

Все nditer примеры, которые изменяют значения, должны использовать эту запись [...]=.

Я не рекомендую использовать nditer, по крайней мере, не код Python. В коде Python это наиболее полезно как способ тестирования идей, которые будут реализованы в вашем собственном скомпилированном коде (с cython). Это не дает никаких преимуществ в скорости.

0 голосов
/ 31 октября 2018

Я предпочитаю numpy.ndindex , чтобы сэкономить некоторые вложенные циклы и сохранить структуру функций на разумном уровне. У меня такое ощущение, что numpy.nditer более полезен для случаев с большим количеством массивов для циклического повторения: 

mat = np.random.rand(2, 3, 4) * 100 - 50
for i in np.ndindex(*mat.shape):
    if mat[i] < -20:
        mat[i] = -20
    if mat[i] > 20:
        mat[i] = 20  
mat /= 20.0

Альтернативой является использование numpy.where для простых условий и операций. В этом случае форма массива не имеет значения: 

mat = np.random.rand(2, 3, 4) * 100 - 50
#              Condition, True, False
mat = np.where(mat < -20,  -20,  mat)
mat = np.where(mat > +20,   20,  mat)
mat /= 20.0
...