关于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!!!