生成指定长度的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 构造稀疏行：

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位互为补集”这一更强对称性。

# 编码 # 排列 # numpy # 您的 # 长度为 # 的是 # 穷举 # 这一 # 推荐使用 # 适用于 # 自定义 # 为其 # 所示

返回目录在线咨询

上一篇：如何优化服务器性能以支持多个高流量网站？
下一篇：C语言中时间的基本用法小结

您的项目需求

*请认真填写需求信息，我们会在24小时内与您取得联系。

全网整合营销服务商

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

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

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

? 总结与建议

您的项目需求