ValueError: operands could not be broadcast together with shapes (3000,20) (20,20)

来源:互联网 发布:vb中len函数的使用方法 编辑:程序博客网 时间:2024/06/06 15:43

本人出现的问题是,两个数组A,B的大小分别为(3000,20) (20,20), 是<numpy.ndarray>类型,而不是<matrix>类型,直接进行乘积C = A*B, 之后,提示上述错误,原因是数组大小“不一致”, 解决方案,不用"*"符号,使用numpy中的dot()函数,可以实现两个二维数组的乘积,或者将数组类型转化为矩阵类型,使用"*"相乘,具体如下:

[python] view plaincopy在CODE上查看代码片派生到我的代码片
  1. A = numpy.array([1,2], [3,4], [5,6])    # 3*2 numpy.ndarray  
  2. B = numpy.array([1,2,3], [4,5,6])       # 2*3 numpy.ndarray  
  3. C = A * B    # raise error: ValueError: operands could not be broadcast together with shapes (3,2) (2,2)  
  4.   
  5. # solution method 1  
  6. C = numpy.dot(A, B)             # C: 3*3 numpy.ndarray  
  7.   
  8. # solution method 2  
  9. A_mat = numpy.matrix(A)  
  10. B_mat = numpy.matrix(B)  
  11.   
  12. C = A_mat * B_mat           # C: 3*3 numpy.matrixlib.defmatrix.matrix  
0 0
原创粉丝点击