我希望有一个像 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)
有没有更好的方法来实现这个?例如内置函数(我没找到任何一个),或者一种在某些方面更快的不同定义。