尋找數組中第二小的元素

import random

arr = [random.randint(0, 10) for i in range(10)]


def two_min_num(arr):
    if len(arr) <= 1:
        return arr
    min_num = arr[0]
    two_num = arr[1]
    if min_num > two_num:
        min_num, two_num = two_num, min_num
    for i in arr:
        if i < min_num:
            two_num = min_num
            min_num = i
        elif min_num < i < two_num:
            two_num = i

    return min_num, two_num


print(arr)
print(two_min_num(arr))