如何返回列表中多个项目的位置？ （Python）

Question

我正在使用一个大数据集（大约 3600 个 x 值和 3600 个 y 值），并且尝试返回与已从原始数据中提取的 y 值相对应的某些 x 值的位置。

for n in new_y:
    if new_y in y:
        new_x.append(index(y))
print(new_x)

我得到的错误代码指出： :43: DeprecationWarning: 元素比较失败；这会在将来引发错误。 if new_y in y:

编辑：我应该提到，在打印 new_x 时，返回一个空列表（[]）

Answer 1

您应该使用 enumerate：

new_x = []
for i, n in enumerate(new_y):
    if n in y:
        new_x.append(i)
print(new_x)

或者，更简洁地说：

new_x = [i for i, n in enumerate(new_y) if n in y]

Answer 2

使用 numpy 库，您可以对数据进行切片，而无需迭代“数据集”。