Python numpy 判断零矩阵,判断某个值(a)是否在矩阵中

系统 1493 0

Python numpy 判断0 矩阵,判断某个值(a)是否在矩阵中

百度找 这个需求, 发现没有一种比较简单的解决方式;很多给的答案是要用 循环的方式解决,我觉得不好。
所以写了 一个简单的方式。

            
              
                # 定义一个矩阵,判断是否是零矩阵
              
              
                import
              
               numpy 
              
                as
              
               np
A
              
                =
              
              np
              
                .
              
              zeros
              
                (
              
              
                (
              
              
                600
              
              
                ,
              
              
                600
              
              
                )
              
              
                )
              
              
                """
numpy 中有个where 的功能,是发现某个值(a)得位置。如果没发现返回空矩阵
"""
              
              
                #判断是否是零矩阵
              
              
                if
              
               np
              
                .
              
              where
              
                (
              
              A
              
                !=
              
              
                0
              
              
                )
              
              
                [
              
              
                0
              
              
                ]
              
              
                .
              
              shape
              
                [
              
              
                0
              
              
                ]
              
              
                ==
              
              
                0
              
              
                :
              
              
                print
              
              
                (
              
              
                'this is a zeros matrix'
              
              
                )
              
              
                else
              
              
                :
              
              
                print
              
              
                (
              
              
                'this is not a zeros matrix'
              
              
                )
              
            
          
            
              
                # 定义一个矩阵,某个值是否在其中
              
              
                import
              
               numpy 
              
                as
              
               np
A
              
                =
              
              np
              
                .
              
              zeros
              
                (
              
              
                (
              
              
                600
              
              
                ,
              
              
                600
              
              
                )
              
              
                )
              
              
a
              
                =
              
              
                1
              
              
A
              
                [
              
              
                300
              
              
                ,
              
              
                300
              
              
                ]
              
              
                =
              
              
                1
              
              
                """
numpy 中有个where 的功能,是发现某个值(a)得位置。如果没发现返回空矩阵
"""
              
              
                #判断是否是零矩阵
              
              
                if
              
               np
              
                .
              
              where
              
                (
              
              A
              
                ==
              
              a
              
                )
              
              
                [
              
              
                0
              
              
                ]
              
              
                .
              
              shape
              
                [
              
              
                0
              
              
                ]
              
              
                ==
              
              
                0
              
              
                :
              
              
                print
              
              
                (
              
              
                '%d is not in your matrix'
              
              
                %
              
              a
              
                )
              
              
                else
              
              
                :
              
              
                print
              
              
                (
              
              
                'the first location of value %d in your matrix is (%d,%d)'
              
              
                %
              
              
                (
              
              a
              
                ,
              
              np
              
                .
              
              where
              
                (
              
              A
              
                ==
              
              a
              
                )
              
              
                [
              
              
                0
              
              
                ]
              
              
                [
              
              
                0
              
              
                ]
              
              
                ,
              
              np
              
                .
              
              where
              
                (
              
              A
              
                ==
              
              a
              
                )
              
              
                [
              
              
                1
              
              
                ]
              
              
                [
              
              
                0
              
              
                ]
              
              
                )
              
              
                )
              
            
          

所以 这个值有几个 还需要再循环么???

np.where 是极好用的工具


更多文章、技术交流、商务合作、联系博主

微信扫码或搜索:z360901061

微信扫一扫加我为好友

QQ号联系: 360901061

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描下面二维码支持博主2元、5元、10元、20元等您想捐的金额吧,狠狠点击下面给点支持吧,站长非常感激您!手机微信长按不能支付解决办法:请将微信支付二维码保存到相册,切换到微信,然后点击微信右上角扫一扫功能,选择支付二维码完成支付。

【本文对您有帮助就好】

您的支持是博主写作最大的动力,如果您喜欢我的文章,感觉我的文章对您有帮助,请用微信扫描上面二维码支持博主2元、5元、10元、自定义金额等您想捐的金额吧,站长会非常 感谢您的哦!!!

发表我的评论
最新评论 总共0条评论