Python使用numpy生成仅含0和1的随机数组或列表

Evan

发布时间：2026-10-13| 6人浏览过

原创

本文介绍如何利用Python的numpy库生成仅包含0和1元素的随机数组。通过np.random.randint函数，可指定输出维度，如一维数组、多维矩阵或单个随机值。示例展示了不同参数配置下的结果，适用于需要二进制随机数据的场景。

具体实现方式如下所示：

>>> import numpy as np
>>> myarray= np.random.randint(0,2,10)输出只含0,1元素的一维数组,长度为10
>>> myarray
array([1, 1, 1, 0, 1, 0, 0, 1, 1, 0])
>>> myarray= np.random.randint(0,2,(3,10))输出只含0,1元素的3行10列数组
>>> myarray
array([[0, 1, 0, 1, 1, 1, 1, 0, 0, 0],
  [0, 1, 1, 1, 1, 0, 1, 0, 1, 0],
  [0, 0, 0, 1, 0, 1, 0, 1, 1, 0]])
>>> myarray=np.random.randint(0,2)当第三个参数值省略时,输出只有一个值,随机为0,1
>>> myarray
0
>>> myarray=np.random.randint(0,2)
>>> myarray
0
>>> myarray=np.random.randint(0,2)
>>> myarray
0
>>> myarray=np.random.randint(0,2)
>>> myarray
1
>>>

以上内容介绍了Python生成仅含0和1元素的随机数组或列表的方法，供读者参考。