KNN算法：'int'对象不可订阅

Question

我已经有一段时间没有使用python并且无法找到解决此问题的方法。我试图改变欧几里德距离函数的等式，但它没有任何好处。也许我无视我的问题的解决方案。

这是我的代码：

from math import sqrt
import csv
from random import shuffle
import numpy as np
import numpy
import matplotlib.pyplot as plt
import operator
import math


iris = datasets.load_iris() 

X = iris.data
y = iris.target



def euclideanDistance(id1, id2):
    for x in range(len(id1)-1):
        dist = np.sqrt(np.sum((int(id2[x]) - int(id1[x]))**2))
    return dist

data1 = [2, 2, 2, 'a']
data2 = [4, 4, 4, 'b']
distance = euclideanDistance(data1, data2)
print(distance)

def mykNN(X, y, x_):
    distance = []
    neighbour = []

    for i in range(len(X)):
        d = euclideanDistance(X[i], x_ )
        distance.append((X[i], d))
    distance.sort(key=operator.itemgetter(1))

    for r in range(k):
        options.append(distance[r][0])
    options = neighbour
    return neighbour

k=3

y_ = mykNN(X, y,k)
print(y_)

无论我如何改变我的功能，它都会出现这个错误。

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-85-2a5bfc4a796d> in <module>
     42 k=3
     43 
---> 44 y_ = mykNN(X, y,k)
     45 print(y_)
     46 

<ipython-input-85-2a5bfc4a796d> in mykNN(X, y, x_)
     31 
     32     for i in range(len(X)):
---> 33         d = euclideanDistance(X[i], x_ )
     34         distance.append((X[i], d))
     35     distance.sort(key=operator.itemgetter(1))

<ipython-input-85-2a5bfc4a796d> in euclideanDistance(id1, id2)
     18 def euclideanDistance(id1, id2):
     19     for x in range(len(id1)-1):
---> 20         dist = np.sqrt(np.sum((int(id2[x]) - int(id1[x]))**2))
     21     return dist
     22 

TypeError: 'int' object is not subscriptable

我很感激你的反应，因为这一直困扰着我。

谢谢。

Answer 1

好吧，错误告诉你，在euclideanDistance()中，id1或id2（或两者）都是一个整数，因为那些是你在该行上索引的两个标识符。通过以下方式：

• 你设置k = 3
• 你打电话给mykNN(X, y, k)，这意味着在mykNN()，x_ == 3。
• 你打电话给euclideanDistance(X[i], x_)，这意味着在euclideanDistance()，id2 == 3。
• 您尝试在指示的行上索引id2。整数不能被索引，因此是例外。

这就是造成你的错误的原因。由于我不确定您的代码到底在做什么，我不能直接推荐修复。