python - 使用掩码替换numpy中的字符串

Question

如果我有一个看起来像 ar 的数组，那么它是一个 numpy 数组？如何使用 numpy 掩码将所有“replaceme”替换为 2012。

ar = [['replaceme', 1,2], ['replaceme', 3,4],['replaceme', 5,6]]

ar = [[2012, 1,2], [2012, 3,4],[2012, 5,6]]

Answer 1

这不是掩码的工作，而是高级索引：

>>> import numpy as np
>>> ar = np.array([['replaceme', 1,2], ['replaceme', 3,4],['replaceme', 5,6]], dtype=object)
>>> ar
array([[replaceme, 1, 2],
       [replaceme, 3, 4],
       [replaceme, 5, 6]], dtype=object)
>>> ar == "replaceme"
array([[ True, False, False],
       [ True, False, False],
       [ True, False, False]], dtype=bool)

所以我们需要做的就是：

>>> ar[ar == "replaceme"] = 2012
>>> ar
array([[2012, 1, 2],
       [2012, 3, 4],
       [2012, 5, 6]], dtype=object)