当前位置:首页 » 《关于电脑》 » 正文

从Python到NumPy,细说最接近人类思维的in操作_Python作业辅导员

15 人参与  2021年04月23日 10:03  分类 : 《关于电脑》  评论

点击全文阅读


在Python语言中,in是一个使用频率非常高的操作符,用于判断对象是否位于字符串、元组、列表、集合或字典中。in操作和人的思维方式高度吻合,写起来近乎于自然语言,充分体现了Python的哲学理念。

>>> 'or' in 'hello world'
True
>>> 5 in {1,2,3,4}
False
>>> 'age' in {'name':'Mike', 'age':18}
True

有趣的是,除了R、javascript、SQL外,包括C/C++在内的主流语言几乎都不支持in操作。这或许可以解释为什么Python语言被认为是最容易学习的编程语言。

习惯了使用Python的in操作符,有时就会自然而然地应用到NumPy数组操作上。比如,下面的写法看起来没有任何问题。

>>> import numpy as np
>>> a = np.arange(9)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8])
>>> 5 in a
True
>>> 10 in a
False

不过,当我尝试在np.where()函数中使用in操作符的时候,出现了意外。

>>> np.where(a>5)
(array([6, 7, 8], dtype=int64),)
>>> np.where(a%2==0)
(array([0, 2, 4, 6, 8], dtype=int64),)
>>> np.where(a in [2,3,5,7])
Traceback (most recent call last):
  File "<pyshell#111>", line 1, in <module>
    np.where(a in [2,3,5,7])
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

使用a>5或者a%2==0作为条件,np.where()函数没有问题,但是,使用a in [2,3,5,7],np.where()就会抛出异常。即便写成下面这样,也不能得到期望得结果。

>>> np.where(a in np.array([2,3,5,7]))
(array([], dtype=int64),)

难道NumPy不支持两个数组之间的in操作吗?不,强大到宇宙无敌的NumPy,怎么会不支持数组之间的in操作呢?NumPy不但支持,而且支持得很好。

>>> p = np.array([2,3,5,7]) # 质数数组
>>> np.in1d(a, p) # 返回a的每一个元素是否是质数的布尔数组
array([False, False,  True,  True, False,  True, False,  True, False])
>>> np.where(np.in1d(a, p)) # 返回数组a中质数的索引序号
(array([2, 3, 5, 7], dtype=int64),)
>>> np.where(np.in1d(a, p), -1, a) # 返回数组a中的质数全部替换为-1的结果
array([ 0,  1, -1, -1,  4, -1,  6, -1,  8])

np.in1d()的参数如果是多维数组,将被自动扁平化,且返回的布尔数组也是扁平化的一维数组。

>>> np.in1d(a.reshape((3,3)), p)
array([False, False,  True,  True, False,  True, False,  True, False])

如果np.in1d()的参数是多维的,且期望返回和原数组结构相同的布尔数组,则应使用np.isin()函数。

>>> np.isin(a.reshape((3,3)), p)
array([[False, False,  True],
       [ True, False,  True],
       [False,  True, False]])
>>> np.where(np.isin(a.reshape((3,3)), p))
(array([0, 1, 1, 2], dtype=int64), array([2, 0, 2, 1], dtype=int64))

若是期望得到两个数组的交集而不是交集元素的索引,下面两种方式都可行。

>>> a[np.where(np.isin(a, p))]
array([2, 3, 5, 7])
>>> np.intersect1d(a, p)
array([2, 3, 5, 7])

第二种方式直接使用np.intersect1d()函数得到两个数组的交集,且自动排序。不过,我更喜欢第一种方式。


点击全文阅读


本文链接:http://zhangshiyu.com/post/17867.html

数组  操作  质数  
<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

最新文章

  • 整容99次,归来做未婚妻后爸大结局_姜茜柳眉姜小姐抖音热门_小说后续在线阅读_无删减免费完结_
  • 朝徽宜燕澹容是什么小说(难搞!燕总又争又抢)(朝徽宜燕澹容)全本完整清爽版在线+无广告结局
  • 重生后我去扫厕所,鉴宝女神却疯了阅读_帝王小姐陈列精校文本_小说后续在线阅读_无删减免费完结_
  • 鹤不归顾宁喧林绾绾完结篇(顾宁喧林绾绾)全篇免费版在线+无障碍结局
  • 终章小说简明月顾烬深完结篇(你走后月光沉入深海)已更新+延伸(简明月顾烬深)清爽版
  • 我死后前夫疯魔了全书阮星舟霍景时在线
  • 救命!我被痴女包围了!列表_救命!我被痴女包围了!(神岛裕树)
  • 被亲姐封印百年,她死后我覆灭三界优质全文_封印小老仙老夫新上热文_小说后续在线阅读_无删减免费完结_
  • 全文七零香江,玄学大佬靠算命发家致富(柳玄黄招娣)列表_全文七零香江,玄学大佬靠算命发家致富
  • 七零娇娇美又媚,冷硬糙汉逃不掉续集(纪淮宋安宁)全本完整免费版_起点章节+后续(七零娇娇美又媚,冷硬糙汉逃不掉)
  • 重活一世,我钓到了真正的顾家家主热门推荐_顾星伯父顾星承阅读_小说后续在线阅读_无删减免费完结_
  • 重生之团长老公才是白月光小说(蒋横波姚沐兰)外篇+结局(重生之团长老公才是白月光)全篇在线阅读

    关于我们 | 我要投稿 | 免责申明

    Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1