首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

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

在Numpy中,你可以使用numpy.arange()函数和numpy.full()函数来有效地分配预定义范围内的值

例如,假设你想要创建一个形状为(5, 5)的数组,并用0到9之间的整数填充它:

代码语言:javascript
复制
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)

输出将是:

代码语言:javascript
复制
[[ 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]]

如果你想要在给定范围内随机分配值,可以使用numpy.random.randint()函数:

代码语言:javascript
复制
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)

输出将类似于:

代码语言:javascript
复制
[[ 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]]

这样,你就可以在Numpy中有效地分配预定义范围内的值了。

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

没有搜到相关的沙龙

领券