如何使用 numpy 在数组中使用 mod

Question

假设我有一个数组

a = ([6,8,10,13,15,18,21])

我还有一个数组

b= ([2,5])

我想返回一个数组，它给我 a%b 的非零值。如果 a mod 中的任何值 b 中的任何值都为零，我不想返回它。

c = ([13,21])

使用numpy.mod(a,b) 返回

ValueError：操作数无法与形状一起广播

我该如何执行？

Answer 1

问题是指numpy无法对给定形状的数组应用np.mod操作，一种解决方案是reshape，例如：

import numpy as np

a = np.array([6, 8, 10, 13, 15, 18, 21]).reshape((-1, 1))
b = np.array([2, 5])

print(a[np.mod(a, b).all(1)].reshape(-1))

输出

[13 21]

请注意，您需要重新整形才能获得请求的输出。最好的解决方案是@PaulPanzer 提出的：

import numpy as np

a = np.array([6, 8, 10, 13, 15, 18, 21])
b = np.array([2, 5])

print(a[np.mod.outer(a, b).all(1)])

输出

[13 21]

进一步

1. 关于 numpy 广播，请参阅 1 和 2。
2. 开启outer。