In [1]:

    

import numpy as np


    

In [5]:

    

bb = np.random.normal(size=(5, 2))


    

In [8]:

    

a = np.array([0,0,1,0,1])


    

In [27]:

    

bb


    

    
Out[27]:





array([[-0.08819615,  1.0877351 ],
       [-0.90845974,  0.83847354],
       [-2.6116481 , -0.5365784 ],
       [ 1.07325898,  0.68744885],
       [-1.51767685, -0.17879938]])

In [30]:

    

bb


    

    
Out[30]:





array([[-0.08819615,  1.0877351 ],
       [-0.90845974,  0.83847354],
       [-2.6116481 , -0.5365784 ],
       [ 1.07325898,  0.68744885],
       [-1.51767685, -0.17879938]])

In [44]:

    

np.diag([1,1,1,1,1]).dot(bb)


    

    
Out[44]:





array([[-0.08819615,  1.0877351 ],
       [-0.90845974,  0.83847354],
       [-2.6116481 , -0.5365784 ],
       [ 1.07325898,  0.68744885],
       [-1.51767685, -0.17879938]])

In [17]:

    

a


    

    
Out[17]:





array([0, 0, 1, 0, 1])