https://researchdatapod.com/python-indexerror-too-many-indices-for-array/