函数为某些数组返回错误的值

Question

我最近开始编写代码，正在编写一个函数，该函数返回给定数组中的3个最大数字。该函数适用于大多数阵列，但不适用于其他阵列。代码如下：

def findThreeLargestNumbers(array):
  list = []
  if len(set(array)) != 1:
    while len(list) <= 2:
      for element in array:
        if element == max(array):
          list.append(element)
          array.remove(element)
    list.reverse()
    return list
  else:
    for element in array:
      newlist = [element, element, element]
      return newlist

例如，当我输入数组[1,2,3,4,5]时，函数返回[3,4,5]。但是，当我输入[55,43,11,3，-3,10]时，函数返回4个值：[10,11,43,55]

为什么会这样

非常感谢你

Answer 1

在for的结尾添加一个中断

  def findThreeLargestNumbers(array):
      list = []
      if len(set(array)) != 1:
        while len(list) <= 2:
          for element in array:
            if element == max(array):
              list.append(element)
              array.remove(element)
              break
        list.reverse()
        return list
      else:
        for element in array:
          newlist = [element, element, element]
          return newlist