在Numpy中有效地分配预定义范围内的值

import numpy as np

# 创建一个形状为(5, 5)的空数组
arr = np.empty((5, 5), dtype=int)

# 使用arange()函数和full()函数填充数组
for i in range(arr.shape[0]):
    arr[i] = np.full(arr.shape[1], i * arr.shape[1])

print(arr)

[[ 0  0  0  0  0]
 [ 5  5  5  5  5]
 [10 10 10 10 10]
 [15 15 15 15 15]
 [20 20 20 20 20]]

import numpy as np

# 创建一个形状为(5, 5)的空数组
arr = np.empty((5, 5), dtype=int)

# 使用randint()函数填充数组
arr = np.random.randint(0, 10, size=(5, 5))

print(arr)

[[ 3  7  2  8  4]
 [ 9  1  6  3  0]
 [ 2  7  5  1  8]
 [ 4  3  9  6  7]
 [ 8  0  4  9  5]]