Python 在Numpy中合并两个相同形状数组的最干净方法
我们有一些形状相同的数组,希望合并它们 所谓“合并”,我的意思是输出一个新数组,每个位置的每个数组中的每个I,j的和Python 在Numpy中合并两个相同形状数组的最干净方法,python,multidimensional-array,numpy,Python,Multidimensional Array,Numpy,我们有一些形状相同的数组,希望合并它们 所谓“合并”,我的意思是输出一个新数组,每个位置的每个数组中的每个I,j的和 import numpy as np first = np.array([[1,1],[1,1]]) second = np.array([[2,2],[2,2]]) third = np.array([[3,3],[3,3]]) 结果应该是: [[6,6], [6,6]] 这是我的代码…但有更干净的方法吗?我似乎找不到内置的方法: def merge_arrays(arra
import numpy as np
first = np.array([[1,1],[1,1]])
second = np.array([[2,2],[2,2]])
third = np.array([[3,3],[3,3]])
结果应该是:
[[6,6],
[6,6]]
这是我的代码…但有更干净的方法吗?我似乎找不到内置的方法:
def merge_arrays(arrays):
output = arrays[0]
for a in arrays[1:]:
for i,row in enumerate(a):
for j,col in enumerate(row):
output[i,j] += a[i,j]
return output
merge_arrays([first, second, third])
只是
output=first+second+third
或np.sum([first,second,third],axis=0)
该死的,你赢了我几秒钟,因为你知道答案,而我不得不在ipython试一试:D@KlausWarzecha-如果我不先测试,我通常会被烧伤!(这一次我也这么做了……当我第一次键入它时,忘记了轴=0)@Joe Kington是的,我看到你在编辑,而我的简单加法还在工作,还在为np.sum()的语法感到困惑