Python 如何使用另一个数据阵列屏蔽一个数据阵列

我有一个

项目

数组，它具有形状

（n，3）

，还有一个

计数

数组，它具有相同的形状

（n，3）

。如何使

items

中的每个

C

都具有

count=0

而不使用循环

items = np.array([['B', 'A', 'C'],
    ['B', 'B', 'C'],
    ['A', 'B', 'A'],
    ['C', 'C', 'C'],
    ['B', 'B', 'B']])

counts = np.array([[1, 3, 2],
    [4, 2, 3],
    [2, 2, 1],
    [3, 2, 1],
    [1, 2, 1]])

预期产出：

>>> counts
np.array([[1, 3, 0],
    [4, 2, 0],
    [2, 2, 1],
    [0, 0, 0],
    [1, 2, 1]])

---

您需要的是：

counts[items == 'C'] = 0

更明确的表达方式是：

c_indices = np.where(items == 'C')
counts[c_indices] = 0

---

没有充分的理由为此使用

where

。它更长，更慢，消耗更多的内存，在我看来更不清晰+我想知道如何制作面具，真不敢相信这么简单。哦，好吧