如題目所示，需要尋找數(shù)組中的第二大和第三大的數(shù)值，不能直接排序，因?yàn)橹苯訉?duì)數(shù)組進(jìn)行排序后取數(shù)值會(huì)忽略存在多個(gè)最大值的情況。因此需要對(duì)數(shù)組進(jìn)行去重，有兩種方法，一種是采用集合的方式去重，一種是采用np.unique()進(jìn)行去重。

1. 使用集合set()去重+排序sorted()

arr = [1, 2 ,3, 3, 5, 5, 6, 7, 7]
print(sorted(set(arr))[-2])
6

2. 使用np.unique()+排序sorted()

import numpy as np

arr = [1, 2 ,3, 3, 5, 5, 6, 7, 7]
print(sorted(np.unique(arr))[-2])
6

3. 二者的效率比較

不過針對(duì)規(guī)模較大的數(shù)組，需要比較一下兩者運(yùn)行的效率。

使用np.unique()+排序sorted()，針對(duì)長(zhǎng)度為10000的數(shù)組，耗時(shí)15秒

import numpy as np
from timeit import Timer

def test():
    return sorted(np.unique([i for i in range(10000)]))[-2]


if __name__ == '__main__':
    t1 = Timer("test()", "from __main__ import test")
    print(t1.timeit(10000))

# 輸出結(jié)果
15.3470649

使用集合set()去重+排序sorted()，針對(duì)長(zhǎng)度為10000的數(shù)組，耗時(shí)5秒

import timeit

print(timeit.timeit(stmt="sorted(set([i for i in range(10000)]))[-2]", number=10000))

# 輸出結(jié)果
5.7564598

可見使用set()集合去重的效率更高一些。

撒花~