r - 如何使用 ifelse() 命令在 R 中创建虚拟变量

Question

我正在尝试为 R 创建一个虚拟变量。问题是在我的餐厅“类型”数据集下有许多分类变量。其中，我希望素食餐厅的值为 1，其余值为 0。因此，当我运行回归摘要时，我得到截距，b1 为 reviews_number，b2 为素食餐厅。例如，非素食餐厅为 y=b0+b1(reviews_number)，素食餐厅为 y=b0+b1(reviews_number)+b2(Vegan)。提示是使用 ifelse() 命令，但我似乎无法将系数简化为 3。否则，我需要为每种类型的餐厅分别创建一个值......

Answer 1

假设您的数据框名为 df，您可以使用以下方法创建虚拟变量 (Vegan):

df$Vegan <- ifelse(df$type == "Vegan", 1, 0) # where variable type is type of restaurants 

但是，您应该注意，如果 type 存储为因子，您还可以使用 y=b0+ 获取每种类型餐厅的系数(与引用水平相比) b1(reviews_number)+b2(type) 即 y~reviews+type，正如@mlt 所指出的。