如何将NumPy数组中的随机单个元素替换为指定值

首先定义目标数组：

import numpy as np
TEST = np.array([[1,2,3,4,5], [0,0,1,1,1], [0,1,2,4,5]])

方法一：直接生成随机行列索引 #

利用数组形状生成随机的行、列下标，通过位置定位元素完成替换：

# 获取数组的行数和列数
rows, cols = TEST.shape
# 生成随机行索引（范围0到rows-1）
rand_row = np.random.randint(rows)
# 生成随机列索引（范围0到cols-1）
rand_col = np.random.randint(cols)
# 替换目标元素
TEST[rand_row, rand_col] = 6

方法二：通过扁平索引转二维索引 #

先将数组视为一维序列，生成随机索引后再转换为原数组的二维下标：

# 获取数组总元素数
total_elements = TEST.size
# 生成随机扁平索引（范围0到total_elements-1）
flat_idx = np.random.randint(total_elements)
# 将扁平索引转换为二维数组的行、列下标
row, col = np.unravel_index(flat_idx, TEST.shape)
# 替换目标元素
TEST[row, col] = 6

两种方法均基于位置索引操作，完全不受数组中重复值的影响，确保只替换随机选中的单个元素。

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