python - 想要连接随机均匀生成的值

Question

我有以下代码，我想为其编写一个函数。输出应该是 x 作为数据帧，y 作为系列，即使数据帧具有 x 和 y 作为列就足够了。

x = np.arange(0,50)
x = pd.DataFrame({'x':x})

# just random uniform distributions in differnt range

y1 = np.random.uniform(10,15,10)
y2 = np.random.uniform(20,25,10)
y3 = np.random.uniform(0,5,10)
y4 = np.random.uniform(30,32,10)
y5 = np.random.uniform(13,17,10)

y = np.concatenate((y1,y2,y3,y4,y5))
y = y[:,None]

我尝试了以下操作，但它没有返回从 y1 到 y5 的值。我的代码是:

x = np.arange(0,50)
x = pd.DataFrame({'x':x})
def colm(p,q):
    s, t = p, q
    a = ['y1', 'y2', 'y3', 'y4', 'y5']
    for i in a:
        i = np.random.uniform(s, t, 10)
        s, t = s+10, t+10
        return i

Answer 1

尝试像这样修改函数:

def colm(p, q, chunk_len=10):
    x = np.arange(0,5 * chunk_len)
    x = pd.DataFrame({'x':x})

    # just random uniform distributions in differnt range

    ys = [np.random.uniform(p_, q_, chunk_len) for p_, q_ in zip(p, q)]

    y = np.concatenate(ys)

    return x, pd.Series(y)

您现在可以将其用作

x, y = colm([10, 20, 0, 30, 13], [15, 25, 5, 13, 17])