Python Check Number Of Elements In Array