Python fof data analysis - chapter 2 - example