python - 如何切片 numpy.zeros

Question

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

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

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

Answer 1

由于它是一个矩阵，您可以使用以下命令将最后三个条目设置为 1

In [1]: a[0, -3:] = 1
Out[1]: array([[ 0.,  0.,  0.,  0.,  0.,  0.,  0.,  1.,  1.,  1.]])