subtract rows one by one from numpy array

Question

I have a 2D numpy array, A.

I want to subtract each rows, one by one, from A, and store the row-wise absolute sum in an array.

Is there a way to carry out the operation without using the for loop ? Below is the code with for loop.

import numpy as np
A = np.random.randint(5,size=(8,9))
b = np.zeros(A.shape[1]);
for i in xrange(A.shape[0]):
    b = b + np.sum(np.absolute(A - A[i,:]), axis=0)

Divakar · Accepted Answer · 2017-01-29 12:12:25Z

1

You could use broadcasting -

(np.abs(A[:,None,:] - A)).sum(axis=(0,1))

Steps :

(1) Keeping the last axis aligned get two versions of A :

Input1 (A[:,None,:])    :  M x 1 x N
Input2 (A)              :      M x N

Get the absolution differences between these two inputs resulting in a 3D array.

(2) Sum along the first two axes for final output.

answered Jan 29, 2017 at 12:12

Divakar

222k19 gold badges273 silver badges374 bronze badges

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

Collectives™ on Stack Overflow

subtract rows one by one from numpy array

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