给定一个 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
等等。
我该怎么做呢?