4 python numpy list subtraction
假设我有:
[1, 2, 3, 4]
和整数
6
我想6
与列表中的每个元素进行比较,并返回绝对值差最小的元素,在本例中为4
。有没有一种有效的Numpy
方法来做到这一点?
您可以使用argmin
绝对差来提取索引,然后可以使用索引来提取元素:
a = np.array([1, 2, 3, 4])
a[np.abs(a - 6).argmin()]
# 4
Run Code Online (Sandbox Code Playgroud)
归档时间: |
|
查看次数: |
2945 次 |
最近记录: |