在NumPy数组中找到与其他值最接近的值

Question

我想在yrs中找到最接近2013（_tmp）的元素。

yrs = [2000.0, 2001.0, 2002.0, 2003.0, 2004.0, 2005.0, 2006.0, 2007.0, 2008.0, 2009.0, 2010.0, 2011.0, 2012.0, 2014.0, 2015.0, 2016.0]
_tmp = 2013
min(yrs, key=lambda x: abs(x - _tmp))

我尝试过列表理解，但它会抛出NameError。我怎么能这样做？也许是numpy？

Answer 1

正如我所提到的，使用np.argmin：

yrs[np.abs(np.array(yrs) - _tmp).argmin()]

• 将yrs转换为numpy.ndarray（如果已经是数组则跳过）
• 从_tmp
中的每个元素广播减去yrs
• np.abs找到绝对值（我们希望绝对最少）
• np.argmin找到最小值的索引
• 索引到yrs以完成工作。

Answer 2

找到与数字最接近的值的另一种方法

yrs = [2000.0, 2001.0, 2002.0, 2003.0, 2004.0, 2005.0, 2006.0, 2007.0, 2008.0, 2009.0, 2010.0, 2011.0, 2012.0, 2014.0, 2015.0, 2016.0]
yr = 2013

# creates a list l, with the difference between input year and items in yrs
# find the element having the minimum difference with the input element (yr)
l = [abs(x-yr) for x in yrs]
if l:
    print(int(yrs[l.index(min(l))]))

>>> 2012