a = np.array([1,2,3])
b = np.array([4,5])
l = [a,b]
我想要一个功能stack_padding
这样:
assert(stack_padding(l) == np.array([[1,2,3],[4,5,0]])
numpy 是否有实现的标准方法
EDIT: l
可能有更多的元素
I think itertools.zip_longest
with fill_value=0
可以为您工作:
import itertools
a = np.array([1,2,3])
b = np.array([4,5])
l = [a,b]
def stack_padding(l):
return np.column_stack((itertools.zip_longest(*l, fillvalue=0)))
>>> stack_padding(l)
array([[1, 2, 3],
[4, 5, 0]])
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)