我有一个带整数数组的列表,其中每个元素的值都为< = 100.我需要弄清楚每个数组中不相等元素之间的最小差异。到目前为止,我有以下内容(item
表示一个数组):
unq = numpy.unique(item)
mind = numpy.amin(
(numpy.append(unq, [999]))
-(numpy.append([-999],unq))
)
使用numpy
我首先得到排序的唯一元素数组。在结尾添加高正数并在开头添加高负数后,我减去这两个数组并得到最小值。
有没有更快的方法呢?
答案 0 :(得分:1)
我认为您的解决方案没问题,除了使用numpy.append
之外,您最好使用np.diff
,例如np.diff(np.unique(a))
:
In [1]: import numpy as np
In [2]: a = np.random.randint(0,100,size=50)
In [4]: np.unique(a)
Out[4]:
array([ 0, 2, 3, 5, 7, 8, 15, 18, 20, 22, 23, 27, 30, 31, 32, 33, 37,
38, 42, 43, 45, 48, 49, 57, 59, 62, 65, 70, 74, 75, 76, 78, 79, 80,
83, 84, 88, 91, 93, 94, 96, 98])
In [5]: np.diff(np.unique(a))
Out[5]:
array([2, 1, 2, 2, 1, 7, 3, 2, 2, 1, 4, 3, 1, 1, 1, 4, 1, 4, 1, 2, 3, 1, 8,
2, 3, 3, 5, 4, 1, 1, 2, 1, 1, 3, 1, 4, 3, 2, 1, 2, 2])
In [6]: np.diff(np.unique(a)).min()
Out[6]: 1