I would like to roll a 2D numpy array, each row being rolled by an amount defined in a 1D array. For example, I have
A=np.array([[1,2,3], [4,5,6],