Python Find Index Of Element In Array With Rows