文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

NumPy 索引和切片的用法总结

2024-12-03 00:55

关注

[[414020]]

索引和切片

您可以使用与切片 Python列表相同的方法,对NumPy数组进行索引和切片。

  1. >>> data = np.array([1, 2, 3]) 
  2.  
  3. >>> data[1] 
  4. >>> data[0:2] 
  5. array([1, 2]) 
  6. >>> data[1:] 
  7. array([2, 3]) 
  8. >>> data[-2:] 
  9. array([2, 3]) 

你可以这样想象:

您可能需要获取数组的一部分或特定数组元素,以便在进一步分析或其他操作中使用。为此,需要对数组进行子集、切片和/或索引。

如果您想从数组中选择满足特定条件的值,那么NumPy很简单。

例如,如果从这个数组开始:

  1. >>> a = np.array([[1 , 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]) 

可以轻松打印数组中小于5的所有值。

  1. >>> print(a[a < 5]) 
  2. [1 2 3 4] 

例如,还可以选择等于或大于5的数字,并使用该条件对数组进行索引。

  1. >>> five_up = (a >= 5) 
  2. >>> print(a[five_up]) 
  3. [ 5  6  7  8  9 10 11 12] 

可以选择可被2整除的元素:

  1. >>> divisible_by_2 = a[a%2==0] 
  2. >>> print(divisible_by_2) 
  3. [ 2  4  6  8 10 12] 

或者可以使用&和|运算符选择满足两个条件的元素:

  1. >>> c = a[(a > 2) & (a < 11)] 
  2. >>> print(c) 
  3. [ 3  4  5  6  7  8  9 10] 

还可以使用逻辑运算符&和 |返回布尔值,指定数组中的值是否满足特定条件。这对于包含名称或其他分类值的数组很有用。

  1. >>> five_up = (a > 5) | (a == 5) 
  2. >>> print(five_up) 
  3. [[False False False False] 
  4.  [ True  True  True  True] 
  5.  [ True  True  True True]] 

还可以使用np.nonzero()从数组中选择元素或索引。

从这个数组开始:

  1. >>> a = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]) 

可以使用np.nonzero()打印元素的索引,例如,小于5:

  1. >>> b = np.nonzero(a < 5
  2. >>> print(b) 
  3. (array([0, 0, 0, 0]), array([0, 1, 2, 3])) 

在本例中,返回了一个数组元组:每个维度一个。第一个数组表示找到这些值的行索引,第二个数组表示找到这些值的列索引。

如果要生成元素所在的坐标列表,可以压缩数组,遍历坐标列表,然后打印它们。例如:

  1. >>> listlist_of_coordinates= list(zip(b[0], b[1])) 
  2.  
  3. >>> for coord in list_of_coordinates: 
  4. ...     print(coord) 
  5. (0, 0) 
  6. (0, 1) 
  7. (0, 2) 
  8. (0, 3) 

还可以使用np.nonzero()打印数组中小于5的元素,并使用:

  1. >>> print(a[b]) 
  2. [1 2 3 4] 

如果要查找的元素在数组中不存在,则返回的索引数组将为空。例如:

  1. >>> not_there = np.nonzero(a == 42) 
  2. >>> print(not_there) 
  3. (array([], dtype=int64), array([], dtype=int64)) 

 

来源:Python与算法社区内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯