当需要从一个2维矩阵中切分出一个子矩阵时，给定了切分的index_list
import numpy as np
a = np.array([
[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12],
[13, 14, 15, 16],
])
index_list = [1, 2, 3]
# 要得到[[6,7],[10,11]]
# 得不到对应的子矩阵
print(a[index_list, index_list])
# 切分子矩阵的正确姿势
print(a[index_list, :][:, index_list])
输出
[ 6 11 16]
[[ 6 7 8]
[10 11 12]
[14 15 16]]
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/133490.html
