NumPy 高级索引

132 阅读1分钟

  NumPy提供了一些高级索引技术,可以通过整数数组或布尔数组来进行索引。这些高级索引技术可以用于获取数组中的特定元素、行或列。

  整数数组索引:可以使用一个整数数组来获取数组中的特定元素。这个整数数组表示要获取的元素的索引位置。示例:

  import numpy as np

  arr=np.array([1,2,3,4,5])

  idx=np.array([0,2,4])#要获取的元素的索引位置

  result=arr[idx]

  print(result)#输出:[1 3 5]

  www.jshk.com.cn/mb/reg.asp?…

  布尔数组索引:可以使用一个布尔数组来获取数组中满足条件的元素。这个布尔数组的长度必须与原始数组相同,其中True表示要获取的元素,False表示不获取。示例:

  import numpy as np

  arr=np.array([1,2,3,4,5])

  mask=np.array([True,False,True,False,True])#满足条件的元素

  result=arr[mask]

  print(result)#输出:[1 3 5]

  整数数组和布尔数组组合索引:可以同时使用整数数组和布尔数组来获取数组中的元素。示例:

  import numpy as np

  arr=np.array([1,2,3,4,5])

  idx=np.array([0,2,4])#要获取的元素的索引位置

  mask=np.array([True,False,True,False,True])#满足条件的元素

  result=arr[idx][mask]

  print(result)#输出:[1 5]

  这些高级索引技术可以应用于多维数组,可以通过在索引中使用多个整数数组或布尔数组来获取多维数组中的特定元素、行或列。

  了解和掌握这些高级索引技术可以帮助您更灵活地操作和处理NumPy数组。可以参考NumPy的官方文档和其他教程来深入学习有关高级索引的知识。

NumPy 高级索引.png