对于多维范围，是否有 Python 的 range(n) 等价物?

2022-01-24 00:00:00 python numpy iteration range

问题描述

在 Python 上，range(3) 将返回 [0,1,2].多维范围是否有等价物?

On Python, range(3) will return [0,1,2]. Is there an equivalent for multidimensional ranges?

range((3,2)) # [(0,0),(0,1),(1,0),(1,1),(2,0),(2,1)]

因此，例如，在基于图块的游戏中循环通过矩形区域的图块可以写成:

So, for example, looping though the tiles of a rectangular area on a tile-based game could be written as:

for x,y in range((3,2)):

请注意，我不是要求实现.我想知道这是否是一种公认​​的模式，以及 Python 上是否有内置函数或其标准/通用库.

Note I'm not asking for an implementation. I would like to know if this is a recognized pattern and if there is a built-in function on Python or it's standard/common libraries.

解决方案

在numpy中是numpy.ndindex.也看看 numpy.ndenumerate.

In numpy, it's numpy.ndindex. Also have a look at numpy.ndenumerate.

例如

import numpy as np
for x, y in np.ndindex((3,2)):
    print(x, y)

这会产生:

0 0
0 1
1 0
1 1
2 0
2 1

相关文章