Question

Oy mates，

我自己正在学习numpy并且很好地处理它，即使在阅读documentation之后，我仍然听不到一些概念。我试图通过这个矩阵，让每一行都有10个一直通过它。

data = np.ones(50).reshape(5,10)
xmax = data.shape[0]
ymax = data.shape[1]
data[range(xmax)::2,range(ymax)] = 10

最后一行代码不正确。我知道如何使用半冒号使用切片 - list[start:stop:step]我知道如何使用花式索引来使用逗号ndarray[range(end1),range(end2)]来完成矩阵，但我如何组合这两种方法呢？

如何使用范围逐步执行多维数组，同时设置启动，停止和步骤？

Answer 1

我认为你想要的是：

>>> data[range(xmax)[::2],:] = 10
>>> data
array([[ 10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.],
       [  1.,   1.,   1.,   1.,   1.,   1.,   1.,   1.,   1.,   1.],
       [ 10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.],
       [  1.,   1.,   1.,   1.,   1.,   1.,   1.,   1.,   1.,   1.],
       [ 10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.,  10.]])'

问题在于：

 >>> range(xmax)::
       File "<stdin>", line 1
        range(xmax)::
                   ^
    SyntaxError: invalid syntax

您需要明确地将切片语法应用于范围：

 >>> range(xmax)[::2]
 [0, 2, 4]

一般参考，你可以这样做：

data[np.arange(start1, end1, step1), np.arange(start2, end2, step2)]

第一个np.arange选择行，第二个np.aranage选择列。

一些可能有用的参考资料：

多维数组，使用范围，同时具有设置开始，停止和步骤？

1 个答案: