寻找子数组在数组中的索引号?

1.np.where语句可以使用
import numpy as np
a=np.array([[1,2,3],
[1,2,1],
[1,1,1]])
print(a)
b=np.array([1,2,1])
print(b)
for i in range(len(a)):
if np.where((a[i,:]==b).all())==np.array([0]):
print(i)
2.使用(a==b).all()判断语句的整体遍历即可
def Index_return(self):
self.errorindex = []
for (i, v) in enumerate(self.errordata):
for (j, v1) in enumerate(self.data):
if (v1 == v).all():
self.errorindex.append(j)
print(self.errorindex)
return self.errorindex
posted @ 2020-07-31 15:49  The-Chosen-One  阅读(235)  评论(0编辑  收藏  举报