我正在尝试编写一个函数,该函数接受一个列表的输入和一个 0 或 1 的选项。如果它是 0,我想返回一个大于绝对值 5 的元素列表。如果它是选项1,我想返回一个元素列表是奇数。我也想使用一个while循环。我哪里错了??
def splitList2(myList, option):
nList = []
element = 0
while element < len(myList):
if option == 0:
if abs(element) > 5:
nList.append(element)
elif option == 1:
if element % 2:
nList.append(element)
element = element + 1
return nList