n_comp = 10
s1 = np.diag(s)
print s1[0].shape
print s1[1].shape
for i in range(s1[0].shape[0]):
    for j in range(s1[1].shape[0]):
        if j >= n_comp:
            s1[i][j] = 0