How to subtract the two different numpy array with the same dimension

Question

I have two NumPy array below:

my_array1 = [np.array([[1, 2, 3],
                      [4, 5, 6],
                      [7, 8, 9]]), 
             np.array([[10],
                      [13],
                      [15]])]
my_array2 = [np.array([[3, 2, 1],
                      [6, 5, 4],
                      [9, 8, 7]]), 
             np.array([[7],
                      [8],
                      [9]])]

I want to calcualte my_array1 - my_array2 as below:

my_want = [np.array([[-2, 0, 2],
                      [-2, 0, 2],
                      [-2, 0, 2]]), 
           np.array([[3],
                     [5],
                     [6]])]

Is there an elegant way to do it in python?

[a1 - a2 for a1, a2 in zip(my_array1, my_array2)] or list(map(operator.sub, my_array1, my_array2)) — Mechanic Pig
– Mechanic Pig, Commented Oct 2, 2022 at 4:33
What I know is [my_array1[i] - my_array2[i] for i in range(len(my_array2))] — Mizzle
– Mizzle, Commented Oct 2, 2022 at 4:34

Алексей Р · Accepted Answer · 2022-10-02 04:35:19Z

2

delta = [i-j for i,j in zip(my_array1,my_array2)]
print(delta)

[array([[-2,  0,  2],
       [-2,  0,  2],
       [-2,  0,  2]]), array([[3],
       [5],
       [6]])]

answered Oct 2, 2022 at 4:35

Алексей Р

7,6272 gold badges9 silver badges19 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

ti7 · Accepted Answer · 2022-10-02 04:33:48Z

0

It looks like not two, but four NumPy arrays, packed into native lists

my_want = []
for index in range(2):
    my_want.append(my_array1[index] - my_array2[index])

>>> my_want
[array([[-2,  0,  2],
       [-2,  0,  2],
       [-2,  0,  2]]), array([[3],
       [5],
       [6]])]

answered Oct 2, 2022 at 4:33

ti7

19.9k8 gold badges50 silver badges87 bronze badges

Collectives™ on Stack Overflow

How to subtract the two different numpy array with the same dimension

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related