毎日Learning

学んだことを共有します

P02 (*) リストの最後の2要素をリストで取得する

def main():
    print my_but_last([]) == None
    print my_but_last("a") == None
    print my_but_last(["a"]) == ["a"]
    print my_but_last(["a", "b"]) == ["a", "b"]
    print my_but_last(["a", "b", "c"]) == ["b", "c"]
    print my_but_last(["a", "b", "c", "d"]) == ["c", "d"]

def my_but_last(lista):
    if not lista or not isinstance(lista, list):
        return None
    if len(lista) == 1:
        return lista
    return lista[len(lista)-2:]

if __name__ == "__main__":
    main()