Python Select Function In R