Python Count Array