Matrix plots | Python