小编Bar*_*sen的帖子

numpy数组打印某个值的索引

给定一个 numpy 数组

A = np.array([[[29, 64, 83],
               [17, 92, 38],
               [67, 34, 20]],
              [[73, 28, 45],
               [19, 84, 61],
               [22, 63, 49]],
              [[48, 30, 13],
               [11, 52, 86],
               [62, 25, 12]]])
Run Code Online (Sandbox Code Playgroud)

我想要某个值的索引,比如 63

值不可能重复或丢失

我做了

idx = np.where(A == 63)

print(idx)
Run Code Online (Sandbox Code Playgroud)

我有

(array([1], dtype=int32), array([2], dtype=int32), array([1], dtype=int32))
Run Code Online (Sandbox Code Playgroud)

我想要的是

[1, 2, 1]
Run Code Online (Sandbox Code Playgroud)

作为列表或其他可迭代的,没有所有这些array, dtype=int32等等。

我该怎么做呢?

python arrays indexing numpy

5
推荐指数
1
解决办法
1万
查看次数

标签 统计

arrays ×1

indexing ×1

numpy ×1

python ×1