Assigning a weighted random variable to a new column in an R dataframe

Question

I have a Dataframe that looks like this in R:

df1

I want to create a new column for either a morning or an afternoon shift based on random weight sampling:

df2

shift	weight
Morning	0.8
Evening	0.2

Is there a way to do this?

df1$shift <- sample(df2, prob = df$weight)

user_3732271 · Answer

We may need to specify the size as the number of rows of 'df1' and replace = TRUE

df1$shift <-  with(df2, sample(shift, prob = weight, 
    size = nrow(df1), replace = TRUE))