diff --git a/function.py b/function.py
index 60936bb..a60e089 100644
--- a/function.py
+++ b/function.py
@@ -63,8 +63,8 @@ def classify(input_vector, dataSet, labels, k=3):
     # 按距离升序排列，选出距离最小的k个点的索引
     nearest_neighbors = distances.argsort()[:k]
 
-    # 获取这k个邻居的标签
-    neighbor_labels = [labels[i] for i in nearest_neighbors]
+    # 获取这k个邻居的标签，并转为列表
+    neighbor_labels = list(labels[nearest_neighbors])
 
     # 返回出现次数最多的标签
     predicted_label = max(set(neighbor_labels), key=neighbor_labels.count)