如何将值作为索引函数分配给numpy数组?

您可以指定一个范围并利用numpy broadcasting :

A = np.zeros((256, 256))
A[:] = range(A.shape[1])
# or A[:] = np.arange(A.shape[1])

您选择的方法将取决于您希望应用的函数。

numpy.mgrid 这似乎是正确的工具(特别是如果你同时需要x和y)。

从文档中复制的示例:

>>> np.mgrid[0:5,0:5]
array([[[0, 0, 0, 0, 0],
        [1, 1, 1, 1, 1],
        [2, 2, 2, 2, 2],
        [3, 3, 3, 3, 3],
        [4, 4, 4, 4, 4]],
       [[0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4],
        [0, 1, 2, 3, 4]]]) 

试试这个:

a=[]
for i in range(4):
    b=[]
    for j in range(4):
        b.append(j)
    a.append(b)