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数组将沿第一维解压缩,因此需要转置数组。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)