如何为DataFrame生成指定离散值的随机列（数值/字符串）

生成Hg列（指定数值随机选取） #

np.random.uniform生成的是连续区间内的均匀随机值，所以会出现0.07、0.12这类不在你指定列表里的数值。要从固定列表里随机选值，用np.random.choice更合适：

import pandas as pd
import numpy as np

# 定义允许的Hg数值列表
hg_candidates = [0.05, 0.10, 0.15, 0.20, 0.25]
# 设定DataFrame行数，可根据实际需求调整
row_count = 10
# 创建包含Hg列的DataFrame
df = pd.DataFrame({'Hg': np.random.choice(hg_candidates, size=row_count)})

生成Outcome列（指定字符串随机选取） #

同样用np.random.choice从指定字符串列表中随机选择，直接添加到已有的DataFrame中：

# 定义允许的Outcome选项
outcome_candidates = ["Positive", "Negative"]
# 添加Outcome列
df['Outcome'] = np.random.choice(outcome_candidates, size=row_count)

示例输出 #

运行上述代码后，会得到类似如下的结果：

Hg    Outcome
0  0.15  Positive
1  0.10  Positive
2  0.20  Negative
3  0.05  Negative
4  0.25  Positive
...  ...        ...

内容的提问来源于stack exchange，提问作者GhettoCoder