使用布尔值索引返回大于或等于该数组最小元素两倍的数组元素

时间:2019-06-15 16:06:26

标签: python numpy

使用布尔索引来返回大于或等于数组最小元素两倍的数组元素。

返回数组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尺寸的矩阵,而不仅仅是矢量?

0 个答案:

没有答案