simultaniously operations in a loop for an arrays of arrays of arrays python

Question

Lets say have an array of the form

r=([[[3,2,1],[5,4,1]],[[10,6,3],[5,3,1]],[[9,5,2],[8,4,1]]])

And I want to do a subtraction between the elements of each array but getting the following array

r=([[[3-3,3-2,3-1],[5-5,5-4,5-1]],[[10-10,10-6,10-3],[5-5,5-3,5-1]],[[9-9,9-5,9-2],[8-8,8-4,8-1]]])

r=([[[0,1,2],[0,1,4]],[[0,4,7],[0,2,4]],[[0,4,7],[0,4,7]]])

I have tried loops inside loops but I don't get what I want because I don't know how to restart the value I'm subtracting in each array.

Is there a reason you're not using numpy? Also, you haven't show your code to get that output for us to assist with. — roganjosh
– roganjosh, Commented May 14, 2018 at 19:57

Kasravnd · Accepted Answer · 2018-05-14 20:05:32Z

1

You can use a nested list comprehension as following:

In [45]: [[[i[0]-j for j in i] for i in sub] for sub in r]
Out[45]: [[[0, 1, 2], [0, 1, 4]], [[0, 4, 7], [0, 2, 4]], [[0, 4, 7], [0, 4, 7]]]

answered May 14, 2018 at 20:05

Kasravnd

108k19 gold badges167 silver badges195 bronze badges

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

Collectives™ on Stack Overflow

simultaniously operations in a loop for an arrays of arrays of arrays python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related