Python Lambda If Function In R