按列解压缩NumPy数组

按列解压缩NumPy数组,第1张

按列解压缩NumPy数组

您可以解压缩数组的转置,以便将列用作函数参数:

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]])

让我们编写一个将列加在一起的函数(通常

x.sum(axis=1)
在NumPy中完成):

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

然后我们有:

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

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



欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5674852.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-17
下一篇 2022-12-17

发表评论

登录后才能评论

评论列表(0条)

保存