绘制给定主题的数据

Question

我使用R和lme函数进行了随机效应的线性回归：

ana1 <- lme(Y ~ 1+X,data,random = list(group=pdDiag(~1+X)))

其中Y是我的观察，X是回归量的矢量，group是一个分组向量（group的每个元素都是1到N之间的数字，表示这是观察所属的主题。）

在进行线性回归后，我可以使用以下方法获得拟合值：

fit <- fitted(ana1)

我想要做的是：创建一个函数，其中包含参数ana1和k，一个介于1和N之间的数字，以及绘制（在同一个图上）主题k的观察结果和拟合值（用不同的颜色）。

我不知道该怎么做。我觉得可以使用predict完成此操作。

Answer 1

plot(Y~X,data) #To plot all data
points(data$X[k],predict(ana1,data[k,]),col='red') #add a predicted point in red