b = np.random.randint(0, 10, (4, 3))
print(b)
"""[[5 2 2]
[1 3 0]
[8 9 0]
[1 4 8]]"""
print(np.where(b>5))
## (array([2, 2, 3], dtype=int64), array([0, 1, 2], dtype=int64))
print(*np.where(b>5))
## [2 2 3] [0 1 2]
np.where和*np.where---numpy
最新推荐文章于 2022-10-15 21:48:28 发布