diag4Darray function

Define function to create a 'diagonal' array or get the diagonal of an array