Question

假设我有以下列表：

l = ["watermelon", "banana", "orange", "apple"]

我想编写一个函数，该函数返回列表中另一个元素之前的元素。例如：

>> is_before("banana", "watermelon")
>> False
>> is_before("banana", "apple")
>> True

编写这种功能的最Python方式是什么？

Answer 1

您可以这样做（假设没有重复项）：

l = ["watermelon", "banana", "orange", "apple"]

indeces = {w: i for i, w in enumerate(l)}

def is_previous(x, y):
    return indeces[x] < indeces[y]

>>> is_previous("banana", "watermelon")
False
>>> is_previous("banana", "apple")
True

这不能处理任何参数不在初始列表中的情况。

Answer 2

mylist = ["watermelon", "banana", "orange", "apple"]

def is_before(prev_item, target, arr):
    return prev_item in arr[:arr.index(target)]

>>>is_before("banana", "apple", mylist)
True
>>>is_before("banana", "watermelon", mylist)
False

如果要处理重复项，可以使用类似的内容

def find_item_last_index(count, item, arr, index=0):
    # A recursive function for finding the last index of an item in a list
    if count == 1:
        return index + arr.index(item)
    return (find_item_last_index(count-1, item, arr[arr.index(item)+1:],  
                                 index+arr.index(item)+1))

def is_before(prev_item, target, arr):
    return prev_item in arr[: find_item_last_index(arr.count(target), target, arr)]

mylist =  ["watermelon", "apple", "banana", "orange", "apple"]

>>>is_before("banana", "apple", mylist)
True
>>>is_before("banana", "watermelon", mylist)
False

查找列表中两个元素之间关系的最pythonic方法

2 个答案: