来自字符串列表的子数组

Question

我只是想在两个列表中提取它们等于某个字符串的索引。

说：

a = ['foo' for _ in range(5)]
a.extend(['bar' for _ in range(5)])  
print(a)
['foo', 'foo', 'foo', 'foo', 'foo', 'bar', 'bar', 'bar', 'bar', 'bar']
b = ['foo' for _ in range(3)]
b.extend(['bar' for _ in range(7)])
print(b)
['foo', 'foo', 'foo', 'bar', 'bar', 'bar', 'bar', 'bar', 'bar', 'bar']

然后我只想要列表 a 中带有 'foo' 和列表 b 中带有 'bar' 的所有索引，这应该是 [3, 4]

index = (a == 'foo') & (b == 'bar')

不像处理 numpy 数组那样工作。我该如何进行这项工作？ 多谢！！ ？？？

Answer 1

用enumerate函数迭代，过滤掉所有foo的元素，只得到它们的索引，像这样

>>> [index for index, item in enumerate(a) if item == "foo"]
[0, 1, 2, 3, 4]

---

编辑后，问题发生了巨大变化。但是这个解决方案会起作用

>>> a = ['foo', 'foo', 'foo', 'foo', 'foo', 'bar', 'bar', 'bar', 'bar', 'bar']
>>> b = ['foo', 'foo', 'foo', 'bar', 'bar', 'bar', 'bar', 'bar', 'bar', 'bar']
>>> [idx for idx, values in enumerate(zip(a, b)) if values == ("foo", "bar")]
[3, 4]