Python Subset Rows In R