整数数组中非等元素之间的最小差异

时间:2013-03-04 09:17:00

标签: python numpy

我有一个带整数数组的列表,其中每个元素的值都为< = 100.我需要弄清楚每个数组中不相等元素之间的最小差异。到目前为止,我有以下内容(item表示一个数组):

unq  = numpy.unique(item)
mind = numpy.amin(
        (numpy.append(unq, [999]))
       -(numpy.append([-999],unq))
       )

使用numpy我首先得到排序的唯一元素数组。在结尾添加高正数并在开头添加高负数后,我减去这两个数组并得到最小值。

有没有更快的方法呢?

1 个答案:

答案 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