Python: help needed with numpy.diag function

Question

I have a numpy matrix M with variations along the rows and columns. The column data from M I want to place on the diagonal of a new matrix P. For example:

for i i