diff --git a/function.py b/function.py
index 4e59d37..a6820de 100644
--- a/function.py
+++ b/function.py
@@ -41,6 +41,7 @@ name = "梅晓东"
 
 
 # 这里写自己的代码实现
+import operator
 def classify0(inX, dataSet, labels, k=0):
     """
     KNN 分类算法。