我想用numpy创建这种数组:
[[[0,0,0], [1,0,0], ..., [1919,0,0]],
[[0,1,0], [1,1,0], ..., [1919,1,0]],
...,
[[0,1019,0], [1,1019,0], ..., [1919,1019,0]]]
Run Code Online (Sandbox Code Playgroud)
我可以通过以下方式访问:
>>> data[25][37]
array([25, 37, 0])
Run Code Online (Sandbox Code Playgroud)
我试过用这种方式创建一个数组,但它并不完整:
>>> data = np.mgrid[0:1920:1, 0:1080:1].swapaxes(0,2).swapaxes(0,1)
>>> data[25][37]
array([25, 37])
Run Code Online (Sandbox Code Playgroud)
你知道怎么用numpy解决这个问题吗?