Yes, this is how you do this, but in python
https://doc.rust-lang.org/std/iter/trait.Iterator.html#method.count