https://researchdatapod.com/python-typeerror-only-size-1-arrays-can-be-converted-to-python-scalars/