np.where()[0] 和 np.where()[1]

最新推荐文章于 2023-11-08 14:45:01 发布

Zzzz26~

最新推荐文章于 2023-11-08 14:45:01 发布

阅读量2k

点赞数 5

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

本文链接：https://blog.csdn.net/ysh1026/article/details/109559981

版权

Python 专栏收录该内容

5 篇文章 1 订阅

订阅专栏

import numpy as np

a = np.arange(12).reshape(3,4)
print('a:', a)
print('np.where(a > 5):', np.where(a > 5))
print('a[np.where(a > 5)]:', a[np.where(a > 5)])
print('np.where(a > 5)[0]:', np.where(a > 5)[0])
print('np.where(a > 5)[1]:', np.where(a > 5)[1])
print(a[np.where(a > 5)[0], np.where(a > 5)[1]])

a: [[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
np.where(a > 5): (array([1, 1, 2, 2, 2, 2]), array([2, 3, 0, 1, 2, 3]))
a[np.where(a > 5)]: [ 6  7  8  9 10 11]
np.where(a > 5)[0]: [1 1 2 2 2 2]
np.where(a > 5)[1]: [2 3 0 1 2 3]
[ 6  7  8  9 10 11]

np.where()[0] 表示行索引，np.where()[1]表示列索引