Separate Variable of Matix (Get Basis)

edit

I try to look for basis of Matrix group center. However, I have problem to separate variable. For simplicity, I have

A=Matrix([a1,a1],[0,a2])

Then, I have to get matrix [[1,1],[0,0]] and [[0,0],[0,1]] which linear combination of matrix above. Could anyone help me? Thanks