如何在两个numpy阵列的尺寸不匹配时回收元素？

  • Z时代
  • 2024-01-10
  • 分类：问答

我想做一些类似于以下R代码numpy，其中y是回收利用。如何在两个numpy阵列的尺寸不匹配时回收元素？

R> x=rbind(c(1,2,3), c(4,5,6)) 
R> y=c(1,2) 
R> x/y 
    [,1] [,2] [,3] 
[1,] 1 2.0 3 
[2,] 2 2.5 3

显然，以下代码不适用于numpy。有人知道什么是可用的等效Python代码吗？谢谢。

>>> x=numpy.array([[1,2,3], [4, 5, 6]]) 
>>> y=numpy.array([1,2]) 
>>> x/y 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
ValueError: operands could not be broadcast together with shapes (2,3) (2,)

回答：

如何

x=numpy.array([[1,2,3], [4, 5, 6]]) 
y=numpy.array([1,2]) 
x/y[:, None]

y[:, None]接通(2,)阵列到阵列(2,1)，从而允许分割的广播与x。

以上是 如何在两个numpy阵列的尺寸不匹配时回收元素？ 的全部内容， 来源链接： utcz.com/qa/267021.html

回到顶部