Python np.where()矩阵索引

自我o救赎

已于 2023-03-13 12:22:00 修改

阅读量2.1k

点赞数

分类专栏： Python语言专栏文章标签： python

于 2022-04-20 15:25:39 首次发布

本文链接：https://blog.csdn.net/y15520833229/article/details/124298764

版权

Python语言专栏专栏收录该内容

70 篇文章 3 订阅

订阅专栏

import numpy as np

A=np.array([[1,3,5],[2,4,6],[9,8,7]])

print(np.where(A==1))
(array([0], dtype=int64), array([0], dtype=int64))

print(np.where(A==2))
(array([1], dtype=int64), array([0], dtype=int64))

A=np.array([[1,3,5],[2,4,6],[9,2,7]])

print(np.where(A==2))
(array([1, 2], dtype=int64), array([0, 1], dtype=int64))

A=np.array([[1,3,2],[2,4,6],[9,2,7]])

print(np.where(A==2))
(array([0, 1, 2], dtype=int64), array([2, 0, 1], dtype=int64))