Smart handling of Python array with many indices

Question

I have the following piece of code:

p1 = np.array([[[[[[[[[[0.]*2]*2]*2]*2]*2]*2]*2]*2]*2]*2)
s = [0]*10
#
# Do something with s
#
p1[s[0],s[1],s[2],s[3],s[4],s[5],s[6],s[7],s[8],s[9]] += 1

Is there a smarter way of:

Creating p1 without all those brackets, and
Access the components of p1 with a string or a list?

I have in mind something like:

p1[s] += 1

or:

p1[*s] += 1

For example, what if instead of 10 indices I wanted N indices?

Dan D. · Accepted Answer · 2015-04-24 21:43:55Z

7

np.array([[[[[[[[[[0.]*2]*2]*2]*2]*2]*2]*2]*2]*2]*2)

is better written as:

np.zeros((2,2,2,2,2,2,2,2,2,2))

Or as there are ten 2s:

np.zeros((2,)*10)

edited Apr 24, 2015 at 21:43

answered Apr 24, 2015 at 21:41

Dan D.

75k15 gold badges111 silver badges129 bronze badges

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

2 Comments

geodude Over a year ago

Thanks a lot! Do you also know anything about the second problem?

nickie Over a year ago

For the second problem, with this solution or any other, just use p1[s]. Just make sure that s is of the right type, e.g., a tuple here.

Collectives™ on Stack Overflow

Smart handling of Python array with many indices

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related