Python Count Elements In Array With Function