0

0

生成指定长度的0-1组合数组:构造每行含x个1和x个0的所有唯一排列

心靈之曲

心靈之曲

发布时间:2025-12-25 23:02:02

|

445人浏览过

|

来源于php中文网

原创

生成指定长度的0-1组合数组:构造每行含x个1和x个0的所有唯一排列

本文介绍如何使用numpy与itertools高效生成所有长度为2x、且恰好包含x个1和x个0的二进制行向量组成的二维数组,适用于组合枚举、约束排列生成等场景。

您提供的示例中,当 x=2 时输出4行、每行4列(即 2*x=4),且每行恰好含两个1和两个0;同理,x=3 时每行6列、含三个1和三个0。但需注意:原始答案中使用的 np.column_stack([a, 1 - a]) 实际生成的是「前x位与后x位互补」的结构(如 [0,1,1,0]),这虽满足总长2x、总和为x,但并非穷举所有含x个1的组合——它仅覆盖了“前半段任意、后半段取反”的子集(共 2^x 种),而完整组合数应为 C(2x, x)(例如 x=3 时为 C(6,3)=20,但示例只列出8行)。

因此,正确解法应分为两类需求:

✅ 需求一:生成所有长度为 2x、恰含 x 个 1 的二进制向量(完整组合)

推荐使用 itertools.combinations 枚举 1 的位置索引,再用 NumPy 构造稀疏行:

千图设计室AI助手
千图设计室AI助手

千图网旗下的AI图像处理平台

下载
import numpy as np
from itertools import combinations

def binary_arrays_with_x_ones(x):
    n = 2 * x
    # 生成所有从 [0, 1, ..., 2x-1] 中选 x 个位置置1的组合
    ones_positions = list(combinations(range(n), x))
    # 初始化全零数组
    arr = np.zeros((len(ones_positions), n), dtype=int)
    # 向每行对应位置填1
    for i, pos in enumerate(ones_positions):
        arr[i, list(pos)] = 1
    return arr

# 示例:x = 2 → 输出 6 行(C(4,2)=6),非原题所示4行
print(binary_arrays_with_x_ones(2))
# [[1 1 0 0]
#  [1 0 1 0]
#  [1 0 0 1]
#  [0 1 1 0]
#  [0 1 0 1]
#  [0 0 1 1]]
⚠️ 注意:原问题示例输出仅含4行(x=2)或8行(x=3),实为 “前x位任意,后x位为前x位的按位取反” 的特殊结构(即 a + (1-a) 拼接),其本质是 2^x 种模式,而非 C(2x,x)。若此结构即您的真实需求,则原始答案正确:

✅ 需求二:生成“前x位任意、后x位为其逻辑反”的数组(即 a | (1-a) 拼接)

from itertools import product
import numpy as np

def binary_arrays_complementary(x):
    # 生成所有长度为x的0-1组合
    a = np.array(list(product([0, 1], repeat=x)))
    # 拼接:前x位为a,后x位为1-a(逐元素取反)
    return np.hstack([a, 1 - a])

print(binary_arrays_complementary(2))
# [[0 0 1 1]
#  [0 1 1 0]
#  [1 0 0 1]
#  [1 1 0 0]]

该结果与您原始答案一致,但顺序和数值与题干示例不完全匹配(题干以 [1,0,1,0] 开头,而此处以 [0,0,1,1] 开头)。可通过排序或自定义排列调整,例如按字典序逆序或按特定模式筛选。

? 总结与建议

  • 若目标是数学意义上的所有含x个1的2x维二进制向量 → 用 combinations(range(2*x), x)(推荐,完备且直观);
  • 若目标是前后对称互补结构(如编码中的自反码) → 用 product([0,1], repeat=x) 拼接取反;
  • 避免 np.meshgrid 或盲目 product(*[[0,1]]*2x)(会生成 2^(2x) 行,远超需求,且需后过滤);
  • 对于大 x(如 x > 15),内存敏感场景建议使用生成器替代 list(...),或借助 numba/dask 加速。

最终选择取决于您的实际约束条件——请优先确认:是否必须严格满足“每行1的数量 = x”,还是仅需满足“前x位 + 后x位互为补集”这一更强对称性。

相关专题

更多
高德地图升级方法汇总
高德地图升级方法汇总

本专题整合了高德地图升级相关教程,阅读专题下面的文章了解更多详细内容。

68

2026.01.16

全民K歌得高分教程大全
全民K歌得高分教程大全

本专题整合了全民K歌得高分技巧汇总,阅读专题下面的文章了解更多详细内容。

123

2026.01.16

C++ 单元测试与代码质量保障
C++ 单元测试与代码质量保障

本专题系统讲解 C++ 在单元测试与代码质量保障方面的实战方法,包括测试驱动开发理念、Google Test/Google Mock 的使用、测试用例设计、边界条件验证、持续集成中的自动化测试流程,以及常见代码质量问题的发现与修复。通过工程化示例,帮助开发者建立 可测试、可维护、高质量的 C++ 项目体系。

54

2026.01.16

java数据库连接教程大全
java数据库连接教程大全

本专题整合了java数据库连接相关教程,阅读专题下面的文章了解更多详细内容。

39

2026.01.15

Java音频处理教程汇总
Java音频处理教程汇总

本专题整合了java音频处理教程大全,阅读专题下面的文章了解更多详细内容。

19

2026.01.15

windows查看wifi密码教程大全
windows查看wifi密码教程大全

本专题整合了windows查看wifi密码教程大全,阅读专题下面的文章了解更多详细内容。

85

2026.01.15

浏览器缓存清理方法汇总
浏览器缓存清理方法汇总

本专题整合了浏览器缓存清理教程汇总,阅读专题下面的文章了解更多详细内容。

20

2026.01.15

ps图片相关教程汇总
ps图片相关教程汇总

本专题整合了ps图片设置相关教程合集,阅读专题下面的文章了解更多详细内容。

11

2026.01.15

ppt一键生成相关合集
ppt一键生成相关合集

本专题整合了ppt一键生成相关教程汇总,阅读专题下面的的文章了解更多详细内容。

47

2026.01.15

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 4万人学习

Rust 教程
Rust 教程

共28课时 | 4.5万人学习

Git 教程
Git 教程

共21课时 | 2.8万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号