numpy的陣列和從基質中提取柱，不同的形狀

Question

我試圖做numpy的積分：

A = n.trapz(B,C) 

但是我有一些問題，B和C形狀

B是一個填充

：陣列numpy的零inizialized起作用

B=np.zeros((N,1)) 

C是從基質中提取的列，總是與numpy的inizialized

C = D[:,0] 
D = np.zeros((N,2)) 

的問題是：

n.shape(B) # (N,1) 
n.shape(C) # (N,) 

我怎麼能管理這個？

Answer 1

嘗試

B = np.zeros(N) 
np.trapz(B, C) 

此外，您np.trapz接受多維數組，所以形狀(N, 1)的陣列都OK;你只需要指定一個軸來正確處理它。

B = np.zeros((N, 1)) 
C = D[:, 0] 
np.trapz(B, C.reshape(N, 1), axis=1)