https://manishbarnwal.com/posts/subsetting_dataframe_python/