Python-numpy-高级索引

最新推荐文章于 2025-06-05 11:54:14 发布

Yolandera

最新推荐文章于 2025-06-05 11:54:14 发布

阅读量4.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： python bumpy 索引

本文链接：https://blog.youkuaiyun.com/Yolandera/article/details/80311664

Python 专栏收录该内容

31 篇文章

订阅专栏

这篇博客探讨了在numpy的ndarray中进行高级索引的方法，包括整数索引和布尔索引。在整数索引中，行和列索引可以结合使用，甚至可以结合切片操作。布尔索引则利用布尔值来选取特定元素，例如与NaN相关的处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在做张量运算的时候经常会用到索引,这里记录一下numpy包中的ndarray中的高级索引
1 整数索引
对于矩阵,有行索引,列索引

import numpy as np

a = np.array([[1,2,3], [4,5,6],[7,8,9]])
b = a[[0,2],[0,2]]
c = a[1,2]
print(a)
print(b)
print(c)
#输出
[[1 2 3]
 [4 5 6]
 [7 8 9]]
[1 9]
6

注意: 对多个数据进行索引时,将行索引放一起,列索引放一起.

import numpy as np

a = np.array([[1,2,3], [4,5,6],[7,8,9]])
rows = np.array([[0,0],[2,2]])
cols = np.array([[0,2],[0,2]])
b = a[rows,cols]
print(b)
#输出
[[1 3]
 [7 9]]

有点意思啊,这个格式返回的还是一个ndarray.

还可以借助切片:或…与索引数组组合.如下面例子

import numpy as np

a = np.array([[1,2,3], [4,5,6],[7,8,9]])
b = a[1:3, 1:3]
c = a[1:3,[1,2]]
d = a[...,1:]
print(b)
print(c)
print(d)
#输出
[[5 6]
 [8 9]]
[[5 6]
 [8 9]]
[[2 3]
 [5 6]
 [8 9]]

2 布尔索引

import numpy as np

a = np.array([[1,2,3], [4,5,6],[7,8,9]])
b = a[a > 4]
print(b)
#输出
 [5 6 7 8 9]

第二个例子:

import numpy as np

a = np.array([np.nan,  1,2,np.nan,3,4,5])
b = a[~np.isnan(a)]
print(b)
#输出
[ 1.  2.  3.  4.  5.]

关于NaN,以及与None的区别我在另一篇博客中有写.
这里用了取补符号 ~