python – 返回一些元素的函数返回np.array的副本

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 返回一些元素的函数返回np.array的副本，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1153字，纯文字阅读大概需要2分钟。

内容图文

我有一个Numpy数组和一个索引列表,以及一个包含需要进入这些索引的值的数组.

我知道如何实现这一目标的最快方式是：

In [1]: a1 = np.array([1,2,3,4,5,6,7])

In [2]: x = np.array([10,11,12])

In [3]: ind = np.array([2,4,5])

In [4]: a2 = np.copy(a1)

In [5]: a2.put(ind,x)

In [6]: a2
Out[6]: array([ 1,  2, 10,  4, 11, 12,  7])

请注意,我必须复制a1.我正在使用的是包装一个以数组作为输入的函数,所以我可以将它提供给一个优化器,它将改变一些这些元素.

所以,理想情况下,我希望能够在一行中返回原始文件的修改副本,其行为如下：

a2 = np.replace(a1, ind, x)

原因是我需要像这样应用它：

def somefunction(a):
    ....

costfun = lambda x: somefunction(np.replace(a1, ind, x))

使用a1和ind常数,这将给我一个仅仅是x的函数的成本函数.

我目前的后备解决方案是自己定义一个小函数：

def replace(a1, ind, x):
    a2 = np.copy(a1)
    a2.put(ind,x)
    return(a2)

……但这对我来说并不是很优雅.

=&GT有没有办法把它变成lambda函数？

解决方法:

好吧,你要求一个单行,这里是一个使用稀疏矩阵与Scipy's csr_matrix –

In [280]: a1 = np.array([1,2,3,4,5,6,7])
     ...: x = np.array([10,11,12])
     ...: ind = np.array([2,4,5])
     ...: 

In [281]: a1+csr_matrix((x-a1[ind], ([0]*x.size, ind)), (1,a1.size)).toarray()
Out[281]: array([[ 1,  2, 10,  4, 11, 12,  7]])

内容总结

以上是互联网集市为您收集整理的python – 返回一些元素的函数返回np.array的副本全部内容，希望文章能够帮你解决python – 返回一些元素的函数返回np.array的副本所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端