numpy.ndarray中相应元素的最大值

时间:2013-10-24 04:16:59

标签: python numpy max multidimensional-array

这似乎是一个非常简单的问题,但到目前为止我还没有找到解决方案。

我有两个numpy.ndarrays(比如说​​A,B)并且想要创建具有相同形状和维度的第三个(比如C),这样C中的每个元素都是相应元素的最大值在A和B中。

到目前为止我所尝试的并不起作用,但说实话,我没有尝试过多少(但我没有想法)

In [173]: A
Out[173]: 
array([[  2.12752806e-314,   2.12752806e-314],
       [  2.16171674e-314,   6.32300944e+233]])

In [174]: B
Out[174]: 
array([[  2.13899304e-314,   2.13899304e-314],
       [  2.16168421e-314,   2.78136354e-309]])

In [175]: max(A, B)
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-175-c06ce068ec08> in <module>()
----> 1 max(A, B)

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

2 个答案:

答案 0 :(得分:1)

您正在寻找np.maximum(A,B)

答案 1 :(得分:0)

np.where

怎么样?
In [29]: where(A>B, A, B)
Out[29]: 
array([[  2.13899304e-314,   2.13899304e-314],
       [  2.16171674e-314,   6.32300944e+233]])