Python for循环结果转换成一列

Question

如图，for循环后的结果好像不在同一列里，要怎么把它变成[1,2,3,4,5……]这样一列的形式呀？

Answer 1

ret = []
for i in range(cols): 
     lr =LinearRegression() 
     f=lr.fit(x[i*ys:i*ys+xs],y[i*ys:i*ys+xs]) 
     p=lr.predict(test[xs+i*ys:(xs+ys)+i*ys]) 
     pp= [j for item in p for j in item] 
     ret.extend(pp)
     
print(ret)

这样行不行