Numpy数组和列从矩阵中提取，形状各异

Question

我正在尝试与numpy集成：

A = n.trapz(B,C)

但是我有一些B和C形状的问题

B是一个用numpy零函数

初始化的填充数组

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

C是从矩阵中提取的列，始终使用numpy：

进行初始化

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)的数组是可以的;你只需要指定一个轴来正确处理它。

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