生成指定大小的只含0,1元素的随机数组或列表,一般可以使用Python中的numpy库或random库实现。下面详细介绍两种方法。
方法一:使用numpy库
import numpy as np
# 定义生成大小为(3, 4)的二维随机数组
arr = np.random.randint(0, 2, size=(3, 4))
print(arr)
解释:
- 导入numpy库,可以使用其中的
random
模块生成随机数组; - 使用
np.random.randint()
方法,生成随机0和1之间的整数,size
参数指定数组大小。
运行上面代码,可以得到如下输出:
array([[1, 1, 1, 1],
[1, 1, 1, 0],
[0, 1, 0, 1]])
方法二:使用random库
import random
# 定义生成大小为(3, 4)的二维随机列表
lst = [[random.randint(0, 1) for j in range(4)] for i in range(3)]
print(lst)
解释:
- 导入random库,使用其中的
randint()
方法生成随机整数; - 嵌套循环生成二维列表,外层循环控制行数,内层循环控制列数。
运行上面代码,可以得到如下输出:
[[0, 1, 1, 0],
[1, 0, 0, 0],
[0, 1, 0, 1]]
上述两种方法根据需要生成的数组或列表大小,可以进行相应的调整。其中,使用numpy库生成随机数组的方式更加简单和高效。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:python生成多个只含0,1元素的随机数组或列表的实例 - Python技术站