0
我試圖計算跨矩陣的行並將該值存儲在不同的矩陣中。這是做這件事的最有效的方法,或者是否有我應該知道的內置函數。python numpy:計算跨矩陣的行
import numpy as np
a = np.array([ [1, 2, 3], [4, 5, 6], [7, 8, 9] ])
def calc_across(matrix):
frame = []
for row in matrix:
frame.append([row[0] * row[1]/2. * row[2]/3]) # period present to generate floats
return np.array(frame)
b = calc_across(a)
如果我做print b
我得到以下矩陣:
b = [ [1.], [20.], [84.] ]
如果a
是3×3,b
必須是3×1(3行,1列)。如果a
是10x3,b
必須是10×等