def ref(x,t):
x=np.array(x)
if isinstance(t,int):
y=np.zeros(len(x))
for i in range(len(x)):
if t>i:
y[i]=x[int(i-t[i])]
return y
try:
len(t)
if len(t)==len(x):
y=np.array([x[int(i-t[i])] for i in range(len(t))])
return y
else:
return ref(x,int(t[-1]))
except:
return ref(x,int(t))
这样封装一下
ref(c,1)
ref(c,t)
直接不就跟分析家一样了嘛