numpy - 如何在没有循环的情况下为多个整数类型在 numpy 中创建 bool 索引

Question

我想创建一个数组的 bool 索引，只要给定列表中的任何整数出现在数组中，该索引就会具有真值。现在我通过循环遍历测试整数列表并为每个我按位 or 构造一个单独的 bool 掩码来做到这一点。像这样在一起:

boolean_mask = original_array == list_of_codes[0]
for code in list_of_codes[1::]:
    boolean_mask = boolean_mask | (original_array == code)

是否有任何速记 numpy 符号可以在没有循环的情况下进行等效操作？

Answer 1

您有 np.in1d :

boolean_mask = np.in1d(original_array, list_of_codes)

应该这样做。请注意 np.in1d展平两个阵列，所以如果您的 original_array是多维的，你将不得不重新塑造它，例如:

boolean_mask = boolean_mask.reshape(original_array.shape)