生成随机值并将结果追加到下一列中

Question

我要做的第一件事是从用户那里获得四个数字，并将它们放在第一列中(例如:10、30、60、80)，然后除了第一列之外，我还需要创建另一个列(第二列)，第二列的行应该变化如下所示。

10  Values should range from 1-2
30 values should range from 3-4
60 values should range from 5-8
80 values should range from 3-8

在此基础上，我希望生成1000数据集，并将它们保存为单个文件。预期产出应相似

10   1.2  1.5 ...
30   3.8  3.4 ...
60   5.8  5.2 ...
80   3.8  4.2 ...

剧本：

import numpy as np
import random
inp_data=[10, 30, 60, 80]
random_data=np.random.uniform(1,2)

我被困在这里了，请提前帮助experts.Thanks。

Answer 1

你可以用熊猫和numpy做这件事

import pandas as pd
import numpy as np

inp_data=[10, 30, 60, 80]

# mapping dict for the ranges
ranges = {10: [1,2],
          30: [3,4],
          60: [5,8],
          80: [3,8]
         }

# initialize random generator
rng = np.random.default_rng()

# load input data as dataframe
df = pd.DataFrame(inp_data, columns=['inp_data'])
# apply random generator to input data column
random_df = df['inp_data'].apply(lambda x: rng.uniform(ranges[x][0], ranges[x][1], size=(1000))).apply(pd.Series)

random_df[:] = np.sort(random_df.values, axis=1) # this sorts the row, set axis=0 to sort column-wise
# join generated data with df and save as csv
df = df.join(random_df.round(3))
df.to_csv('output.csv', index=False, header=False)