带有ifelse()语句的lm()-R

时间:2019-03-05 20:50:58

标签: r formula linear-regression lm

假设我有这个数据框:

df <- data.frame(GN1 = sample(1:10, 10 ,replace=TRUE),
           GN2 = sample(1:10, 10 ,replace=TRUE),
           GN3 = sample(1:10, 10 ,replace=TRUE),
           E10 = sample(1:10, 10 ,replace=TRUE),
           PSV7 = sample(1:10, 10 ,replace=TRUE),
           PEC3 = sample(1:10, 10 ,replace=TRUE),
           PEC4 = sample(1:10, 10 ,replace=TRUE),
           AC6 = sample(1:10, 10 ,replace=TRUE),
           AC7 = sample(1:10, 10 ,replace=TRUE),
           stringsAsFactors = FALSE)

   GN1 GN2 GN3 E10 PSV7 PEC3 PEC4 AC6 AC7
1    7   3  10   6    4    4    3   9   3
2    2   5   6   6    6    6    5   7   1
3    7   6  10   6    9    1    9   7   5
4    7   1   8   9    2    4    5   5   7
5    8   3   3   8    6    8    9   5  10
6    7   1   1   8    9    3    8   9   4
7    4   6   4   7    2    6    9   8   9
8    7   8   8   7    2    1    7   6   5
9    1   9   4   8    5    5    2   7   1
10   4   9   2   1    4    4   10   2   9

我想运行以下公式:

c_SA=lm(formula = GN1 ~ ifelse(df2$if_a == 1,PEC3+PEC4+AC6,GN2+GN3+E10+PSV7+PEC3), data = df)

df2$if_adf的外部值,它只能接受值01df2仅一行)。如上所示,如果df2$if_a == 1我需要运行变量的第一个“包”,而如果它等于0,则需要运行变量的另一个“包”。

我尝试了as.formula()reformulate(),但没有成功:

c_SA=lm(formula = GN1 ~ ifelse(df2$if_a == 1,as.formula(PEC3+PEC4+AC6),as.formula(GN2+GN3+E10+PSV7+PEC3)), data = df)

另外,还有一些类似的问题(123)。但是,它们将data =参数中的数据帧作为子集,因此我需要使formula =参数受外部源值的约束。

有什么建议吗?

1 个答案:

答案 0 :(得分:5)

使用data = data.sort((a, b) => a['DateX'] > b['DateX'] ? -1 : a['DateX'] > b['DateX'] ? 1 : 0)(不是向量化的if ---因为((希望)不使用向量)选择所需的公式,而不是尝试在公式中使用它:

ifelse