如何将长度不一的二维数组填充后堆叠为(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
相关产品推荐
相关产品推荐

