Python枚举numpy中的2D数组

rlms

我想要一个行为类似于的函数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)

有什么更好的方法吗?例如内置函数(找不到)或以某种方式更快的其他定义。

海梅

您要np.ndenumerate

>>> for (x, y), element in np.ndenumerate(np.array([[i for i in "egg"] for j in range(3)])):
...     print(x, y, element)
... 
(0L, 0L, 'e')
(0L, 1L, 'g')
(0L, 2L, 'g')
(1L, 0L, 'e')
(1L, 1L, 'g')
(1L, 2L, 'g')
(2L, 0L, 'e')
(2L, 1L, 'g')
(2L, 2L, 'g')

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

Python 3.4中的Numpy 2D数组

来自分类Dev

在python中对2d numpy数组进行下采样

来自分类Dev

Python:2D numpy数组中峰的正确位置?

来自分类Dev

Python 2D NumPy数组理解

来自分类Dev

Python:从 3d numpy 数组中的 2d 数组访问保存的点

来自分类Dev

根据Python中的变量将n维numpy数组转换为2D数组

来自分类Dev

NumPy:数组和标量列表中的2D数组

来自分类Dev

Python中的2D数组操作

来自分类Dev

在python中填充2D数组

来自分类Dev

Python 中的动态数组 2D

来自分类Dev

如何使用numpy在现有2D数组中添加2D子数组?

来自分类Dev

根据numpy Python中的条件对2D数组进行子集化

来自分类Dev

Python尝试更新2D numpy数组中的值,值不更新

来自分类Dev

根据numpy Python中的条件对2D数组进行子集化

来自分类Dev

Python numpy保留排序的2D数组的索引列表

来自分类Dev

Python-Numpy 2D数组太长,打印异常?

来自分类Dev

Python:提取2D numpy数组的核心

来自分类Dev

Python:将 2D numpy 数组变成字典

来自分类Dev

numpy查找2D数组中的出现次数

来自分类Dev

在numpy中添加2D数组的问题

来自分类Dev

如何在Numpy中更改2D数组元素?

来自分类Dev

根据列表从2D Numpy数组中删除元素

来自分类Dev

如何从2D NumPy数组中获取直方图数据?

来自分类Dev

在numpy中连接两个2D数组

来自分类Dev

numpy查找2D数组中的出现次数

来自分类Dev

numpy:在2d数组中查找边界框

来自分类Dev

如何在熊猫2D数据框中复制Numpy 2D数组

来自分类Dev

使用numpy使用2D数组的值更改3D数组中的子数组

来自分类Dev

从numpy中的3d数组中删除2d子数组