Numpy Permutations of a 3D Matrix

Question

I've got a n x n x 2 matrix which I want to find all possible permutations of without changing the order of elements in the 3rd dimension.
For example, if my matrix was 2 x 2 x 2 and had the following values:

[[[1,2], [3,4]],  
  [5,6], [7,8]]

Then possible permutations would be:

[[[1,2], [3,4]],  
  [7,8], [5,6]]   

[[[3,4], [1,2]],  
  [5,6], [7,8]]   

[[[1,2], [7,8]],  
  [5,6], [3,4]]

etc.

In other words, I want to treat the tuples as single values when finding the permutations.

lynn · Accepted Answer · 2015-07-24 10:24:01Z

1

How about something like:

import numpy as np
import itertools

arr = np.array([[[1,2], [3,4]],
                [[5,6], [7,8]]])

for p in itertools.permutations(arr.reshape(-1, 2)):
    print(np.array(p).reshape(arr.shape))

answered Jul 24, 2015 at 10:24

lynn

10.9k1 gold badge48 silver badges81 bronze badges

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

Collectives™ on Stack Overflow

Numpy Permutations of a 3D Matrix

1 Answer 1

Comments

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Related