NumPy迭代拆分数组后与原数组相等性比较返回False问题求解

• 拆分逻辑与拼接逻辑不匹配
  你生成stream_A时用的是按块拼接的逻辑，stream_A的元素顺序为：前1000个是X的全部元素，接下来1000个是Y的全部元素，再接下来1000个是Z的全部元素，最后1000个是W的全部元素。
  但你的拆分逻辑是：每次从迭代器中取连续4个元素，依次存入4个结果列表，循环1000次。最终result[0]里存储的是stream_A的第0、4、8……3996号元素，完全不是X的连续1000个元素，自然对比不相等，运算结果也会不一致。
• 类型转换的次要影响
  你用dtype=object生成numpy数组时，数组元素是Python对象而非原生numpy浮点类型，直接用==对比也可能出现非预期结果，但这不是核心问题。

方案1：按块拆分（最简便） #

如果要从stream_A还原回X/Y/Z/W四个数组，直接按长度切割即可：

result = np.array_split(stream_A, 4)
# 此时result[0]与X完全相等，result[1]对应Y，以此类推

方案2：按同位置取数（适配逐样本输入需求） #

如果你需要每次取四个数组的同位置元素输入其他函数，直接遍历索引即可，不需要提前拆分：

for i in range(1000):
    x_i, y_i, z_i, w_i = X[i], Y[i], Z[i], W[i]
    # 此处输入你的自定义函数处理

方案3：迭代器实现修正 #

如果你必须用迭代器方式实现拆分，调整迭代逻辑为先取满单个块的所有元素，再取下一个块：

n = 4
block_len = 1000
iter_array = iter(stream_A)
result = []
for _ in range(n):
    current_block = [next(iter_array) for _ in range(block_len)]
    result.append(current_block)

内容的提问来源于stack exchange，提问作者pouchewar