Numpy：将数组中的 500 个随机元素从 0 替换为 1

Question

Numpy：将数组中的 500 个随机元素从 0 替换为 1

import numpy as np
import random

# 2D Grid world of 100 x 100 cells
arr = np.zeros((100,100))

# Arbitrarily change 500 elements from 0 to 1 
for 

Answer 1

您可以选择 0 到 100*100 之间的 500 个随机唯一数字来将您的数组索引为 1D 并分配 1：

import numpy as np

# 2D Grid world of 100 x 100 cells
arr = np.zeros((100,100))

arr.flat[np.random.choice(np.arange(arr.size), 500, replace=False)] = 1

10x10 数组中有 10 个随机 1 的示例：

arr = np.zeros((10,10), dtype=int)

arr.flat[np.random.choice(np.arange(arr.size), 10, replace=False)] = 1

print(arr)

输出：

[[0 0 0 0 0 0 0 0 1 0]
 [0 0 0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0 0 1]
 [0 1 0 0 0 0 0 0 0 1]
 [0 0 0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0 0 0]
 [1 0 0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 1 0 0 1]
 [1 1 0 0 1 0 0 0 0 0]
 [0 0 0 0 0 0 0 0 0 0]]