python - 在列表 A 中找到第一个索引，其中值在列表 B 中

Question

我想在 A 中找到第一个包含 B 值的索引。例如:

A = [1, 'Q', 3, 6, 'R']
B = ['Z', 'I', 'O', 3]
A.function(B)  # would return 2

有没有简单的方法来做到这一点？显然，您可以在 A 中搜索 B 中的每个项目并采用最小索引，但这比尝试每个索引的所有 B 慢.

Answer 1

我认为有两种广泛的方法，具体取决于哪个列表更长:

1. A 比 B 长:

   一次在A 中取一个项目，看看它是否在B 中。重复直到找到第一个。

   next(index for index, item in enumerate(A) if item in B)
   

   如果将 B 转换为 set(感谢 @khelwood)，这会更有效，但请务必在 开始，不在生成器表达式内。

2. B 比 A 长:

   找到 B 中每个项目在 A 中的索引，然后最小化:

   indices = []
   for item in B:
       try:
           indices.append(A.index(item))
       except ValueError:
           pass
   min(indices)
   

   请注意，您不能使用:

   min(index for index in map(A.index, B) if index > -1)
   

   (哎呀!)因为 list 没有一种方法可以无错误地获取索引(参见例如 list.index() function for Python that doesn't throw exception when nothing found )。