Sign Up to our social questions and Answers Engine to ask questions, answer people’s questions, and connect with other people.
Login to our social questions & Answers Engine to ask questions answer people’s questions & connect with other people.
Lost your password? Please enter your email address. You will receive a link and will create a new password via email.
Please briefly explain why you feel this question should be reported.
Please briefly explain why you feel this answer should be reported.
Please briefly explain why you feel this user should be reported.
I have 2 numpy arrays.
One is a 2*2 array.
a = [[1,2],[3,4]]
The other is a 2*2*4 array.
b = [[[0,0,0,0],[0,0,0,0]],[[0,0,0,0],[0,0,0,0]]]
I want to add them so that I have a 2*2*4 array, c.
c = [[[1,0,0,0],[2,0,0,0]],[[3,0,0,0],[4,0,0,0]]]
What’s the correct numpythonic way to do this?
Edit: This appears to work
b[:,:,:1]+=a[:,:,np.newaxis]
Not sure whether you can do the sum in one single steps. Here it is in two steps: