是否有一个numpy函数可以列出numpy.ndarray中所有可能的“坐标”？

它会做我的自定义函数all_possible_coordinates所做的事情：

import numpy as np

def all_possible_coordinates(shape): 
    return np.array([item.ravel() for item in np.meshgrid(*(np.arange(n) for n in shape))]).T

def test_all_possible_coordinates():
    shape = (1, 2, 2)
    assert np.all(all_possible_coordinates(shape)
                  == np.array([[0, 0, 0],
                               [0, 0, 1],
                               [0, 1, 0],
                               [0, 1, 1]]))

test_all_possible_coordinates()

编辑：更优雅，由M Z

def all_possible_coordinates(shape): 
    return np.argwhere(np.ones(shape))