如何在每个组中创建滞后变量? [Matlab的]

时间:2017-01-18 07:06:28

标签: matlab data-processing

我的问题类似于question about creating a lag variable。我需要做的就是在Matlab中实现此代码。

例如,假设我们在Matlab中有以下两个向量:

ID =[2;2;2;2;3;3;5;5]
Pur=[0;1;2;3;1;2;4;5]
[ID,Pur]
ans =

 2     0
 2     1
 2     2
 2     3
 3     1
 3     2
 5     4
 5     5

我正在寻找一种方法来延迟每个ID的Purchase变量。得到类似的东西:

ans =

2     NA
2     0
2     1
2     2
3     NA
3     1
5     NA
5     4

非常感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

这是我找到的一种方式:

ID =[2;2;2;2;3;3;5;5]
Pur=[0;1;2;3;1;2;4;5]

[uniqueValues,~,uniqueIndex] = unique(ID)

n = numel(uniqueValues);  

shop_sess = accumarray(uniqueIndex,1);

shop_sess_cum=cumsum(shop_sess)
shop_sess_r=[0;shop_sess_cum2(1:length(shop_sess_cum)-1)]

lagPur=[]

for i=1:n
shop_sess_r_i=shop_sess_r(i)    
shop_sess_i=shop_sess_cum(i)
temp=Pur(shop_sess_r_i+1:shop_sess_i)
lagtemp=lagmatrix(temp,1)
lagPur=[lagPur;lagtemp]
end