Python生成指定规则01序列时numpy数组维度错误如何修复

原代码问题原因 #

• 元素顺序颠倒：规则要求每个子块为n个0 + (5-n)个1，原代码先写了np.ones(n)，把1放在了0的前面
• 数组未拼接：原代码生成的是包含4个二元组的列表，每个二元组存了两个独立数组，没有把同一子块的0、1数组拼接，也没有把所有子块合并为一个完整序列
• 维度未对齐：拼接后得到的是一维数组，需要额外调整维度为20行1列

修改后可运行代码 #

基础版本（默认浮点类型） #

import numpy as np
# 逐块生成并拼接单个子块的0、1序列
seq_blocks = [np.concatenate([np.zeros(n), np.ones(5 - n)]) for n in range(1, 5)]
# 合并所有子块后调整为20x1维度
result = np.concatenate(seq_blocks).reshape(20, 1)

整数输出版本（和示例格式完全匹配） #

如果需要输出整数类型的0/1，可以指定dtype参数：

import numpy as np
seq_blocks = [np.concatenate([np.zeros(n, dtype=int), np.ones(5 - n, dtype=int)]) for n in range(1, 5)]
result = np.concatenate(seq_blocks).reshape(20, 1)

验证结果 #

运行后：

• 调用result.shape可得到(20, 1)，符合维度要求
• 调用result.flatten()可得到[0 1 1 1 1 0 0 1 1 1 0 0 0 1 1 0 0 0 0 1]，和要求的序列完全一致

内容的提问来源于stack exchange，提问作者W. Walter