add conditional probabilities to the datatable in r

This can be achieved by using the package data.table. The object TITANIC3 is of class data.frame. First you need to convert it to class data.table. When using data.table you can define new columns based on aggregations and a grouping clause directly in one line. Just run the code below.

The new column with the conditional probability of survival is survival_prob. I always recommend using data.table because it is the fastest way to manipulate data in R. However, if you want to proceed your analysis with a data.frame, just use the command setDF(titanic3) to convert the object back to class data.frame.


# convert dataset from data frame to data table 
titanic3 <- copy(TITANIC3)

# define new column survival_prob using by-option
titanic3[, survival_prob := round(100*mean(survived), 1), 
         by = .(fare > 200, pclass, sex)]

CLICK HERE to find out more related problems solutions.

Leave a Comment

Your email address will not be published.

Scroll to Top