python – 按列解压缩NumPy数组

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 按列解压缩NumPy数组，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含560字，纯文字阅读大概需要1分钟。

内容图文

如果我有一个NumPy数组,例如5×3,有没有办法一列一列地解压缩它以传递给函数而不是像这样：my_func(arr [：,0],arr [：,1], arr [：,2])？

有点像* args列表拆包但是按列.

解决方法:

您可以解压缩数组的转置,以便使用函数参数的列：

my_func(*arr.T)

这是一个简单的例子：

>>> x = np.arange(15).reshape(5, 3)
array([[ 0,  5, 10],
       [ 1,  6, 11],
       [ 2,  7, 12],
       [ 3,  8, 13],
       [ 4,  9, 14]])

让我们编写一个函数来将列添加到一起(通常在NumPy中使用x.sum(axis = 1))：

def add_cols(a, b, c):
    return a+b+c

然后我们有：

>>> add_cols(*x.T)
array([15, 18, 21, 24, 27])

NumPy数组将沿第一维解压缩,因此需要转置数组.

内容总结

以上是互联网集市为您收集整理的python – 按列解压缩NumPy数组全部内容，希望文章能够帮你解决python – 按列解压缩NumPy数组所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

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

内容手机端