使用布尔索引来返回大于或等于数组最小元素两倍的数组元素。
返回数组arr中所有大于或等于arr最小元素2倍的元素。
Parameters
----------
arr: NumPy array (n, m)
Returns
-------
NumPy Array, a vector of size between: 0 and (n * m) - 1
我试图弄清楚该如何编写代码,我相信某些代码可以正常工作,但事实并非如此。
import numpy as np
def elements_twice_min(arr):
x = min(arr)
return arr[arr >= 2 * x]
我认为这将返回输入数组中所有元素的最小值的两倍,但是我却收到了以下失败消息:
F
======================================================================
FAIL: test_one (test_methods.TestPython1)
----------------------------------------------------------------------
Traceback (most recent call last):
File "/usr/src/app/test_methods.py", line 21, in test_one
answer))
AssertionError: False is not true
----------------------------------------------------------------------
Ran 1 test in 0.001s
FAILED (failures=1)
也许输入数组是矩阵的问题?它以NumPy数组(n,m)列出参数,我猜这意味着它可以是n x m尺寸的矩阵,而不仅仅是矢量?