https://researchdatapod.com/python-typeerror-only-integer-scalar-arrays-can-be-converted-to-a-scalar-index/