如何基于两个NumPy数组生成符合指定元素重复规则的第三个数组

全程基于NumPy向量化操作，无Python层循环，逻辑通用适配任意行数列数的输入：

• 先拷贝原have数组作为结果基底，所有未被替换的区域直接保留原始值，无需重复计算
• 利用广播机制生成和原数组同形状的布尔掩码，标记每一行需要替换的位置
• 用掩码索引完成批量替换即可

可直接运行的代码 #

import numpy as np

have = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
use = np.array([[2], [3]]) 

# 初始化结果数组，拷贝原始值
result = have.copy()
# 去掉use数组的冗余维度，得到每一行对应的替换长度
replace_len = use.ravel()
# 生成列坐标矩阵，和替换长度做广播比较得到替换掩码
# 掩码为True的位置即属于该行前N个需要替换的列
col_index = np.arange(have.shape[1])
replace_mask = col_index < replace_len[:, None]
# 按行优先顺序生成替换值：每行首元素重复对应长度次数
replace_vals = have[:, 0].repeat(replace_len)
result[replace_mask] = replace_vals

# 输出结果和示例给定的solution完全一致
print(result)

逻辑校验 #

对着示例输入走一遍流程可以验证正确性：

1. 初始result拷贝原始have值，未替换区域的元素完全保留
2. 压平后的replace_len为[2, 3]，对应两行的替换长度
3. 广播比较生成的replace_mask为2行4列的布尔矩阵：

   [[True,  True,  False, False],
    [True,  True,  True,  False]]
   

4. 生成的replace_vals为[1, 1, 5, 5, 5]，按行优先顺序赋值到掩码位置后，第一行第二个元素替换为首值1，第二行第二、三个元素替换为首值5，最终输出和示例完全匹配。

适配说明 #

• 整套逻辑不依赖固定的行、列数，后续输入数组规模变化时无需修改核心逻辑
• 若use数组中的数值超过对应行的总列数，NumPy会自动将整行标记为需要替换的状态，不会触发索引错误；如需严格边界校验，可在生成掩码前增加判断，确保replace_len的所有值不超过对应行的列数。

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