Python列表之元素查找

• • 背景
  • 代码
  • 优化

给定一个元素和经过排序的列表，在列表中查找元素。
如发现元素则返回元素下标，如未发现元素，则按顺序插入元素后返回下标。

lst = [1,3,5,7,9]

num = int(input('输入一个整数：'))

length = len(lst)

if num < lst[0]:
    lst.insert(0, num)
    print('该元素不在给定列表，插入列表后索引为：', lst.index(num))

elif num > lst[length-1]:
    lst.append(num)
    print('该元素不在给定列表，插入列表后索引为：', lst.index(num))

else:
    for i in range(length):
        if lst[i] == num:
            print('该元素已在给定列表，插入列表后索引为：', lst.index(num))
            break
        elif lst[i] < num:
            continue
        else:
            lst.insert(i, num)
            print('该元素不在给定列表，插入列表后索引为：', lst.index(num))
            break

未完待续