python – 是否可以通过一次调用在2维中平铺一个numpy数组

发布时间：2020-12-20 13:42:57 所属栏目：Python 来源：网络整理

导读：我有一个2d numpy数组 a = array([[1,2],[3,4]]) 我想要的是 array([[1,1,2,[1,3,4,4],4]]) 我可以通过2次调用numpy.repeat来完成此操作 repeat(repeat(a,axis=0),axis=1) 但是,有一些参数组合可以通过一次调用来完成吗？解决方法您可以使用numpy.lib.stri

我有一个2d numpy数组

a = array([[1,2],[3,4]])

我想要的是

array([[1,1,2,[1,3,4,4],4]])

我可以通过2次调用numpy.repeat来完成此操作

repeat(repeat(a,axis=0),axis=1)

但是,有一些参数组合可以通过一次调用来完成吗？

解决方法

您可以使用numpy.lib.stride_tricks.as_strided()创建此类数组：

s = min(a.strides)
as_strided(a,shape=(2,2),strides=(2*s,s,0)).reshape(4,4)
#array([[1,#       [1,#       [3,4]])

reshape()正在复制数据,最后创建一个连续的数组.

注意：尽管可以在一次通话中完成,但您的原始解决方案在我的计算机上快4倍.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!