gpt4 book ai didi

Python通过numpy中的二维数组枚举

转载 作者:太空狗 更新时间:2023-10-29 20:43:50 26 4
gpt4 key购买 nike

<分区>

我想要一个行为类似于 enumerate 的函数,但在 numpy 数组上。

>>> list(enumerate("hello"))
[(0, "h"), (1, "e"), (2, "l"), (3, "l"), (4, "o")]

>>> for x, y, element in enumerate2(numpy.array([[i for i in "egg"] for j in range(3)])):
print(x, y, element)

0 0 e
1 0 g
2 0 g
0 1 e
1 1 g
2 1 g
0 2 e
1 2 g
2 2 g

目前我正在使用这个功能:

def enumerate2(np_array):
for y, row in enumerate(np_array):
for x, element in enumerate(row):
yield (x, y, element)

有没有更好的方法来做到这一点?例如。内置函数(我找不到),或在某种程度上更快的不同定义。

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