Assigning a numpy array

Question

I am having trouble creating a numpy array. I have a numpy array with some shape and I am trying to create a new array with the shape (2, other_array_shape). Something like:

import numpy as np
x = np.zeros((100, 100))
y = np.zeros((2, i for i in x.shape))

However, this comes back with invalid syntax error. Can someone tell me how to achieve this?

Ashwini Chaudhary · Accepted Answer · 2014-11-04 13:39:30Z

4

You simply need to concatenate two tuples:

>>> arr = np.zeros((2,) +  x.shape)
>>> arr.shape
(2, 100, 100)

answered Nov 4, 2014 at 13:39

Ashwini Chaudhary

252k60 gold badges478 silver badges519 bronze badges

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

Comments

qasfux · Accepted Answer · 2014-11-04 13:42:51Z

2

You need to append the tuple:

import numpy as np
x = np.zeros((100, 100))
y = np.zeros((2,) + x.shape)

answered Nov 4, 2014 at 13:42

qasfux

2071 silver badge8 bronze badges

Collectives™ on Stack Overflow

Assigning a numpy array

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