Как разделить n-мерный массив по первому значению из измерения - PullRequest
Новая
       13

Как разделить n-мерный массив по первому значению из измерения

1 голос
/ 10 апреля 2020

Учитывая массив измерения N, как разделить все значения в массиве на первое значение из выбранного измерения?

Пример кода: 

import numpy as np
A = np.random.randint(1, 10, size=(3,3,3))
B = A[:,:,0]
C = np.divide(A,B)
A
print()
B
print()
C
print()
C[:,:,0]

Вывод: 

array([[[1, 8, 5],
        [3, 6, 5],
        [5, 4, 2]],

       [[6, 2, 9],
        [4, 2, 2],
        [5, 6, 8]],

       [[3, 3, 1],
        [2, 7, 7],
        [6, 4, 6]]])


array([[1, 3, 5],
       [6, 4, 5],
       [3, 2, 6]])


array([[[1.        , 2.66666667, 1.        ],
        [0.5       , 1.5       , 1.        ],
        [1.66666667, 2.        , 0.33333333]],

       [[6.        , 0.66666667, 1.8       ],
        [0.66666667, 0.5       , 0.4       ],
        [1.66666667, 3.        , 1.33333333]],

       [[3.        , 1.        , 0.2       ],
        [0.33333333, 1.75      , 1.4       ],
        [2.        , 2.        , 1.        ]]])


array([[1.        , 0.5       , 1.66666667],
       [6.        , 0.66666667, 1.66666667],
       [3.        , 0.33333333, 2.        ]])

Я ожидал, что конечный результат C [:,:, 0] будет равен 1. Я предполагаю, что это связано с трансляцией B, но я не думаю, что понимаю, почему он не транслирует B в форму (3,3,3), где он реплицируется вдоль измерения 2.

Ответы [ 2 ]

2 голосов
/ 10 апреля 2020

Чтобы получить ожидаемые результаты, вы можете изменить форму массива B на: 

B = A[:,:,0].reshape(3,-1, 1)

Затем, когда вы разделите, вы получите такой результат, как: 

array([[[1.        , 0.11111111, 0.11111111],
        [1.        , 0.25      , 0.5       ],
        [1.        , 0.88888889, 0.44444444]],

       [[1.        , 0.88888889, 1.        ],
        [1.        , 1.8       , 1.6       ],
        [1.        , 4.5       , 0.5       ]],

       [[1.        , 0.66666667, 0.5       ],
        [1.        , 1.125     , 0.75      ],
        [1.        , 0.5       , 2.25      ]]])

Вы также можете сохранить правильное измерение для вещания, взяв B как: 

B = A[:,:,0:1]
1 голос
/ 10 апреля 2020

Вам нужно изменить форму B так, чтобы она совпала с A[:,:,0]: 

>>> A
array([[[1, 8, 5],
        [3, 6, 5],
        [5, 4, 2]],

       [[6, 2, 9],
        [4, 2, 2],
        [5, 6, 8]],

       [[3, 3, 1],
        [2, 7, 7],
        [6, 4, 6]]])

>>> B = A[:, :, 0]
>>> B
array([[1, 3, 5],
       [6, 4, 5],
       [3, 2, 6]])

# And you need to reorient B as:

>>> B.T[None,:].T
array([[[1],
        [3],
        [5]],

       [[6],
        [4],
        [5]],

       [[3],
        [2],
        [6]]])

>>> A / B.T[None,:].T
array([[[1.        , 8.        , 5.        ],
        [1.        , 2.        , 1.66666667],
        [1.        , 0.8       , 0.4       ]],

       [[1.        , 0.33333333, 1.5       ],
        [1.        , 0.5       , 0.5       ],
        [1.        , 1.2       , 1.6       ]],

       [[1.        , 1.        , 0.33333333],
        [1.        , 3.5       , 3.5       ],
        [1.        , 0.66666667, 1.        ]]])
...