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

Question

11

我有一个numpy结构化数组，其数据类型为：

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

我知道我可以创建一个如下的遮罩：

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

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

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

- jlconlin

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Sven Marnach · Accepted Answer

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

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

注意额外的括号是必需的。