例如，輸入是 2 個 numpy 數組：a = np.array([3, 8, 5])，b = np.array([6, 11, 10, 9, 17])輸出是數組中的元素b，這些元素可以被 中的至少一個元素精確整除a： 6, 10, 9
2 回答
慕娘9325324
TA貢獻1783條經驗 獲得超4個贊
基于 numpy 的解決方案可能如下：
>>> check = b[:, np.newaxis] % a == 0
>>> b[np.any(check, axis=1)]
array([ 6, 10, 9])
慕尼黑5688855
TA貢獻1848條經驗 獲得超2個贊
假設這些數組是一維的，您可以通過簡單的列表理解來完成：
[x for x in b if any([x % y == 0 for y in a])]
添加回答
舉報
0/150
提交
取消