如何在多列上屏蔽 numpy 结构化数组？

Question

我有一个带有 dtype 的 numpy 结构化数组，例如：

A = numpy.empty(10, dtype=([('segment', '<i8'), ('material', '<i8'), ('rxN', '<i8')]))

我知道我可以创建一个掩码，例如：

A[A['segment'] == 42] = ...

有没有办法在多列上创建掩码？例如（我知道这不起作用，但我希望它能起作用）：

A[A['segment'] == 42 and A['material'] == 5] = ...

Answer 1

您可以使用 & 运算符代替 and：

A[(A['segment'] == 42) & (A['material'] == 5)]

请注意，需要额外的括号。