Python如何逐列基于条件判断与随机选择从二维数组生成目标数组

• 选取逻辑错误：未提前筛选当前列中所有值为1的行号，而是随机碰单个行号，未命中1就直接填0，完全不符合“从当前列的有效1值位置中随机选一个”的规则
• 列表追加顺序颠倒：循环内先把上一轮计算的旧值追加进C数组，再计算当前列的取值，导致第一列的结果永远不会进入C，所有元素和列的对应关系错位
• 无有效性兜底：题目已经明确所有列至少有1个1，原代码的0默认值完全是多余的，还会覆盖合法的计算结果

逐列遍历矩阵A：

• 对当前列，先筛出所有元素值为1的行索引列表（题目前提保证这个列表非空，不需要做特殊边界处理）
• 从这个有效行索引列表中随机抽取1个作为选中的行号
• 用行号去数组B中取对应的值，追加到目标数组C中

import numpy as np

# 可替换为你原本的随机生成逻辑，注意生成后校验每列至少存在1个1
A = np.array([[0, 0, 0, 1, 0, 1, 1, 0, 0, 1, 0],
              [0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1],
              [0, 0, 0, 1, 0, 1, 1, 1, 0, 0, 0],
              [1, 1, 0, 1, 0, 0, 0, 1, 1, 1, 0]])
B = ["a1", "b1", "c2", "d2"]
C = []

for col_pos in range(A.shape[1]):
    # 获取当前列所有取值为1的行索引
    valid_row_indices = np.where(A[:, col_pos] == 1)[0]
    # 从有效行中随机选1个
    chosen_row = np.random.choice(valid_row_indices)
    # 存入对应B的取值
    C.append(B[chosen_row])

print(C)

• 按照示例第0列的计算逻辑，筛出来的有效行只有索引3，随机选中后对应B[3]的取值，完全和举例逻辑一致
• 对存在多个1的列（比如第3列所有行都是1），会从所有合法行里随机选取，不存在全0或者无效值的问题

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