diff --git a/function.py b/function.py
index 286bb9a..e6d525e 100644
--- a/function.py
+++ b/function.py
@@ -64,7 +64,7 @@ def classify0(inX, dataSet, labels, k=3):
     # 计算待分类样本与训练集中每个样本的马氏距离
     invCovMat = np.linalg.inv(covMat)
     diffVecs = dataSet - meanVec
-    dist = np.sqrt(np.sum(inX - meanVec, axis=1).dot(invCovMat).dot((inX - meanVec).T))
+    dist = np.sqrt(np.sum((inX - meanVec).dot(invCovMat) * (inX - meanVec), axis=1))
     
     # 获取距离最近的 k 个样本的索引
     kNearestIndices = np.argsort(dist)[:k]