Python中NumPy矩阵[[2,2,0]]索引的含义及原理解析

1. 索引类型与作用 #

这是NumPy里的整数数组索引，核心作用是按你指定的顺序和重复次数，提取原矩阵的行。

2. 每个索引数字的含义 #

NumPy数组的索引从0开始计数：

• 2 对应原矩阵的第3行（即[0, 0, 1, 0]那一行）
• 重复的2 表示再提取一次第3行
• 0 对应原矩阵的第1行（即[1, 0, 0, 0]那一行）

简单说，[2,2,0]就是告诉NumPy：按这个顺序给我拿出这三行。

3. 为什么要用双括号？ #

外层的单括号[]是NumPy数组本身的索引语法（比如identity_matrix[0]里的单括号），内层的[2,2,0]是列表类型的索引器——你需要把这个索引器放在外层的索引括号里，所以形成了[[2,2,0]]的写法。

要是只写identity_matrix[2,2,0]，会被当成多维索引（行2、列2、维度0？显然不符合4×4矩阵的维度），直接报错。

4. 输出结果验证 #

按照上述逻辑，输出就是重复两次第3行、再加上第1行的3×4矩阵：

[[0. 0. 1. 0.]
 [0. 0. 1. 0.]
 [1. 0. 0. 0.]]

内容的提问来源于stack exchange，提问作者gvella29