How to get Matrix diagonal?

Question

How do I get the diagonal of a Matrix into a numpy array?

Answer 1

For PETSc backend and dolfin matrix A:

diagonal = as_backend_type(A).mat().getDiagonal().array

Comments

wrong comment: I found my mistake. It's

.array

and not

.array()