https://csc-ubc-okanagan.github.io/R-Python-Blog/R-Python-Blog/posts/Subsetting-in-Python/