r - 使用 data.table 组与 nls 进行指数曲线拟合
问题描述
我想将指数曲线拟合到下面显示的数据表中的第 1 组和第 2 组,并获得一个新列,其中包含与每个组对应的残差标准误差。指数曲线应遵循y=a*exp(b*x)+c
## Example data table
DT <- data.table(
x = c(1,2,3,4,5,6,7,8,1,2,3,4,5,6,7,8),
y = c(15.4,16,16.4,17.7,20,23,27,35,25.4,26,26.4,27.7,30,33,37,45),
groups = c(1,1,1,1,1,1,1,1,2,2,2,2,2,2,2,2)
但是,我只知道如何使用下面的代码拟合 nls 曲线并获得单个组的残差标准误差,该代码估计了良好的起始参数a、b和c:
subsetDT <- DT[group == 1]
c.0 <- min(subsetDT[,y]) * 0.5
model.0 <- lm(log(y- c.0) ~ x, data=subsetDT)
start <- list(a=exp(coef(model.0)[1]), b=coef(model.0)[2], c=c.0)
model <- nls(y ~ a * exp(b * x) + c,
data = subsetDT, start = start,
control = nls.control(maxiter=500))
sigma <- summary(model)$sigma
我不想DT
在循环中按组进行子集计算sigma
和其他模型信息。
我知道,如果我使用lm
,我将能够执行以下操作来获取包含模型信息的新列:
DT[, `:=` (r.squared=summary(lm(log(y)~x))$r.squared,
int=coef(lm(log(y)~x))[1],
coeff=coef(lm(log(y)~x))[2]
), by=c("groups")]
如何使用:=
来拟合指数曲线并合并我的 nls 参数a、b和c?
解决方案
如果您要在原始数据集中添加 sigma、a、b、c 作为新列,您可以执行以下操作:
DT[, c("sigma", "a", "b", "c") := {
c.0 <- min(y) * 0.5
model.0 <- lm(log(y - c.0) ~ x, data=.SD)
start <- list(a=exp(coef(model.0)[1]), b=coef(model.0)[2], c=c.0)
model <- nls(y ~ a * exp(b * x) + c,
data=.SD,
start=start,
control=nls.control(maxiter=500))
c(.(sigma=summary(model)$sigma), as.list(coef(model)))
},
by=.(groups)]
输出:
x y groups sigma a b c
1: 1 15.4 1 0.2986243 0.5265405 0.4565363 14.56728
2: 2 16.0 1 0.2986243 0.5265405 0.4565363 14.56728
3: 3 16.4 1 0.2986243 0.5265405 0.4565363 14.56728
4: 4 17.7 1 0.2986243 0.5265405 0.4565363 14.56728
5: 5 20.0 1 0.2986243 0.5265405 0.4565363 14.56728
6: 6 23.0 1 0.2986243 0.5265405 0.4565363 14.56728
7: 7 27.0 1 0.2986243 0.5265405 0.4565363 14.56728
8: 8 35.0 1 0.2986243 0.5265405 0.4565363 14.56728
9: 1 25.4 2 0.2986243 0.5265404 0.4565363 24.56728
10: 2 26.0 2 0.2986243 0.5265404 0.4565363 24.56728
11: 3 26.4 2 0.2986243 0.5265404 0.4565363 24.56728
12: 4 27.7 2 0.2986243 0.5265404 0.4565363 24.56728
13: 5 30.0 2 0.2986243 0.5265404 0.4565363 24.56728
14: 6 33.0 2 0.2986243 0.5265404 0.4565363 24.56728
15: 7 37.0 2 0.2986243 0.5265404 0.4565363 24.56728
16: 8 45.0 2 0.2986243 0.5265404 0.4565363 24.56728
推荐阅读
- javascript - 删除元素 vanillaJs
- python - Seaborn 折线图“左后”的问题
- javascript - 检查谷歌表格/谷歌脚本中的日期是否相等
- google-apps-script - 更新 Google 网上论坛发帖权限
- java - 哪种 MIME 类型用于设置 firefox Preference 以自动下载 .ebc 文件?
- r - 就R中的故障步骤而言,For循环与Sapply
- android - 如何在改造时传递带有数组的json对象
- javascript - 从mailto链接中的字段更改?
- javascript - node.js中如何优化array.forEach的计算速度
- groovy - 确定 split() 的正则表达式