https://researchdatapod.com/how-to-solve-python-valueerror-all-the-input-arrays-must-have-the-same-number-of-dimensions/