根据数据帧形状重复变量值

Question

我有一个形状为540,20的变量TS，我的数据帧的形状为660,20。我想按顺序重复TS的值，这样它就可以匹配数据帧的形状，并将其添加为数据帧的最后一列。即使数据帧的形状发生了变化，我也要根据数据帧的大小来调整它的形状。我的解决方案不起作用TS值类似于1.718206999999999951e+01 1.701339000000000112e+01 1.670511400000000179e+01 1.590475700000000003e+01

TS = data[data.keys()[9]] 
TS = np.array(TS)

rpts = np.shape(data)
TS1 = np.repeat(TS,rpts[:1])

Answer 1

您可以在最后一列使用np.tile()，然后在其上使用np.hstack()将其连接到末尾：

import numpy as np
ts = np.array([[1, 2], [4, 5], [5, 6]])
ts1 = np.hstack((ts, np.tile(ts[:, [-1]], 3)))
print(ts)
print(ts1)

这会给你：

[[1 2]
[4 5]
[5 6]]
[[1 2 2 2 2]
[4 5 5 5 5]
[5 6 6 6 6]]