Numpy多维数组索引问题：如何获取array([4,2])格式结果

先解决你的直接需求：要得到 array([4, 2]) 这种Numpy数组格式的结果，有两种简洁的实现方式：

方法1：用Numpy数组构造函数转换 #

把你生成的普通列表传入 np.array()，就能转换成Numpy数组：

Q4 = np.array([Q3[0,1], Q3[0,0]])

方法2：用Numpy高级索引直接提取 #

更贴合Numpy向量式操作逻辑的做法，直接通过索引数组提取目标元素，无需额外转换：

# 提取第0行的第1、第0位元素，直接返回Numpy数组
Q4 = Q3[0, [1, 0]]

Numpy索引操作核心建议 #

• 区分索引返回值类型：单个索引（如 Q3[0,1]）返回Numpy标量，将多个标量放入普通列表只会得到Python列表，必须通过 np.array() 或Numpy原生索引方式生成数组。
• 优先使用Numpy原生索引：提取多个元素时，用索引数组（如 [0, [1,0]]）或切片替代手动拼接列表，既保证返回数组类型，也更高效。
• 灵活选择索引方式：连续元素用切片（如 Q3[0:2, 0]），非连续任意位置元素用索引数组（如 Q3[:, [1,0]] 可直接反转所有列的顺序）。

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