关于AttributeError:'numpy.ndarray' object has no attribute 'index'

使用了一段时间的python，发现一个被忽略的ndarray和list的不同之处。

先上例子：

代码：

import numpy as np

array = [[0,0],[0,1],[0,2],[2,2],[3,3],[4,4]]
Indexs = [1,2,3]
print("first array:",array)

array = np.delete(array,Indexs,0)

print("second array:",array)
print(array.index([0,0]))

运行结果：

first array: [[0, 0], [0, 1], [0, 2], [2, 2], [3, 3], [4, 4]]
second array: [[0 0]
 [3 3]
 [4 4]]

Traceback (most recent call last):
  File "C:\Users\admin\Desktop\meanshift\TEST.py", line 10, in <module>
    print(array.index([0,0]))
AttributeError: 'numpy.ndarray' object has no attribute 'index'

运行结果可以看出，list数组和.ndarray的显示也有不同。
就会报错'numpy.ndarray' object has no attribute 'index'，因为numpy.delete()删除过索引的对象后返回的是ndarray，而不是list，所以不能使用index查找元素的位置。

解决方法：ndarray.tolist()

import numpy as np

array = [[0,0],[0,1],[0,2],[2,2],[3,3],[4,4]]

Indexs = [1,2,3]
print("first array:",array)
array = np.delete(array,Indexs,0)
array = array.tolist()
print("second array:",array)
print(array.index([4,4]))

实验结果：

first array: [[0, 0], [0, 1], [0, 2], [2, 2], [3, 3], [4, 4]]
second array: [[0, 0], [3, 3], [4, 4]]
2

ok!!!