list_numpy = np.array(list_test_2)
list_numpy_index_3 = np.where(list_numpy == element)
list_numpy_index_3_matrix = np.dstack((list_numpy_index_3[0], list_numpy_index_3[1])).squeeze()
# 第三行代码作用是将第二行代码得到的两个array拼接成一个矩阵,一维数组查找不需要这行
>>> print(list_numpy_index_3)
> (array([0, 0, 1, 1, 1], dtype=int64), array([2, 4, 1, 4, 7], dtype=int64))
>>> print(list_numpy_index_3_matrix)
>array([[0, 2],
[0, 4],
[1, 1],
[1, 4],
numpy查找指定元素位置
最新推荐文章于 2024-04-26 00:11:55 发布