Update plot_zhihu.py

Found a better way to count.
Write code more pythonic.