Python Keep Specific Columns In R