如何切片numpy.zeros

时间:2013-10-03 19:03:55

标签: python

a = numpy.zeros((1,10))

所以它是一个10 x 1行向量,我想将最后3个条目设置为1.

我尝试了[: - 3],但这是整个最后一行,假设它是一个矩阵,而不是将其视为该行的最后3个条目。我该怎么做。

1 个答案:

答案 0 :(得分:2)

由于它是一个矩阵,您可以使用以下

将最后三个条目设置为1
In [1]: a[0, -3:] = 1
Out[1]: array([[ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.,  1.,  1.]])