Python Create Array Of Size Nxn