gpt4 book ai didi

python - 获取 N 维数组的所有索引作为列表

转载 作者:行者123 更新时间:2023-11-28 20:57:42 24 4
gpt4 key购买 nike

<分区>

有没有一种方法可以在 Python 中快速高效地获取 N 维数组中所有索引的列表或数组?

例如我们有如下数组的图片:

import numpy as np

test = np.zeros((4,4))

array([[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.],
[0., 0., 0., 0.]])

我想按如下方式获取所有元素索引:

indices = [ [0,0],[0,1],[0,2] ... [3,2],[3,3] ]

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com