关于如何返回列表、数组对象的元素索引问题

1.列表

（1）使用enumerate()函数，返回列表

x = [0, 1, 3, 2, 7, 0, 0]
# 直接输出所有0出现位置的索引
index = [i for i, x in enumerate(pred) if x==0]
print(index)

（2）遍历索引

def get_index(list, item):
	return [i for i in range(len(list)) if list[i] == item]
index = get_index(x, 0)
print(index)

2.数组
（1）可以先把数组对象转化为列表对象，再使用列表对象的方法即可

import numpy as np
ndarray = np.array[1, 2, 4, 2, 5, 0, 0]
list = list(ndarray)

（2）使用numpy模块中的where()或argwhere()函数, 返回数组

index = np.where(ndarray==0)
index = np.argwhere(ndarray==2)
print(index)