最近将matlab的代码迁移到python,虽然有了numpy这个好用的工具,但是因为索引的关系还是很烦。其中matlab find()函数转换成np.where()函数就遇到了一些麻烦。
总结:可以看到在处理一维矩阵时,当find不到要找的值,返回的就是空;当find到了,返回的就是起点为1的索引;
总结:可以看到在处理一维矩阵时,当np.where()不到要找的值,返回的还是一个元组,不过这个元组的第一个元素为空;当np.where()到了,返回的元组的第一个元素,就是起点为0的索引;
总结:find([1,2,3]==value)函数返回的长度,就是矩阵中==value的个数
总结:np.where()返回的是一个元组。可以看到np.where([1,2,3]==value)返回的长度,都是1。即使没有找到value,元组的第一个元素为空,元组自身的长度也是1,这点要特别注意!
那么如何真正的拿到矩阵中对应元素的个数呢?
可以用
len(np.where([1,2,3]==value)[0])
np.where()[0] 表示行的索引,
np.where()[1] 则表示列的索引
总结:可以看到,matlab的find函数的策略是从:第1列由上而下,第2列由上而下…的索引,而且是打平了的索引,不是坐标
即二维坐标:(1,0)
即二维坐标:(0,2)
总结:可以看到,np.where实际上返回的是坐标,并且可作为索引去再找出矩阵a的对应值
参考:
《numpy.where()函数 二维矩阵返回值的解释》
《Python vs Matlab—— find 与 np.where》
《numpy.where() 2种用法》