numpy使用索引从另一个数组赋值

Question

我有这个pointsndarray。在

In [141]: points
Out[141]:
array([[0, 1],
       [2, 3],
       [4, 5],
       [6, 7]])

我有一个分类器，我把这些点分为两类

^{pr2}$

注意，点和i5的长度相同。在

我知道这两个类都有这些值。在

In [143]: c
Out[143]:
array([[2, 3],
       [4, 5]])

我想给我已经分类的值赋值。我期望的最终结果是

In [141]: points
Out[141]:
array([[2, 3],
       [2, 3],
       [4, 5],
       [4, 5]])

如何根据索引在i5上的c对points进行变异/更改？在

Answer 1

只需使用i5作为c上的索引数组，并将c上的索引视图分配给points

import numpy as np

c = np.array([[2, 3],
              [4, 5]])

i5 = np.array([0, 0, 1, 1])

points = c[i5]

# [[2 3]
#  [2 3]
#  [4 5]
#  [4 5]]

注意：根据您对问题的描述，points的初始值似乎并不重要。得出这个结论合适吗？在