You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何将长度不一的二维数组填充后堆叠为(50,917,100)的三维numpy数组

解决方案

你代码报错的核心原因是:numpy数组的append方法和Python列表的append逻辑完全不同,numpy的append操作不会修改原数组,只会返回一个全新的数组,且你已经提前初始化了符合最终形状的全零数组,完全不需要用append实现填充和组装需求。

推荐实现(性能最优)

直接遍历每个sentence,按索引给初始化好的全零数组对应位置赋值即可,剩余位置默认保持0值,自动完成不足长度的填充:

import numpy as np

# 初始化结果数组,dtype指定为和sentence一致的类型,避免数据类型兼容问题
res = np.zeros((50, 917, 100), dtype=all_sentences[0].dtype)

for idx, sentence in enumerate(all_sentences):
    seq_len = sentence.shape[0]
    # 前seq_len个位置赋值为sentence的内容,剩下的保留初始0值
    res[idx, :seq_len, :] = sentence

# 验证结果形状
print(res.shape) # 输出 (50, 917, 100)

可选实现(显式填充后拼接)

如果你需要单独获得每个填充后的sentence数组,可以用np.pad显式填充后再拼接:

padded_list = []
for sentence in all_sentences:
    # 仅对第一维做后端补0,第二维长度已经是100不需要填充
    padded_sent = np.pad(
        sentence,
        pad_width=((0, 917 - sentence.shape[0]), (0, 0)),
        mode="constant",
        constant_values=0
    )
    padded_list.append(padded_sent)

# 沿第0维拼接得到最终数组
res = np.stack(padded_list, axis=0)

内容的提问来源于stack exchange,提问作者whitebear

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.09.23 20:45:01