在数组括号内使用python非零函数

Question

我正在查看链接的GitHub存储库并完成MLA机器学习手册，并偶然发现了这段代码并且不明白这里发生了什么。这里发生了什么？是否有更好，更明确的方式来编写这段python？

https://github.com/KUpypy/Past/blob/fd981bfcc3599914be71d38f0412ef2edbcf6046/MLA/(MLA)%20Chp10/(MLA)%20Chp10_Grouping%20unlabeled%20items%20using%20k-means%20clustering.ipynb

bestClustAss[nonzero(bestClustAss[:,0].A == 1)[0], 0] = len(centList) 
bestClustAss[nonzero(bestClustAss[:,0].A == 0)[0], 0] = bestCentToSplit

此函数抛出以下错误：

local variable 'bestClustAss' referenced before assignment

我的想法是这样的：

if(nonzero(bestClustAss[:,0].A == 1)
     bestClustAss ....... = len(cenList)
if(nonzero(bestClustAss[:,0].A == 0)
     bestClustAss ....... = bestCentToSplit

但是我对于python来说还是太新了，无法自己解决这个问题。