我有两个 numpy 数组:
p_a_colors=np.array([[0,0,0],
[0,2,0],
[119,103,82],
[122,122,122],
[122,122,122],
[3,2,4]])
p_rem = np.array([[119,103,82],
[122,122,122]])
我想从 p_rem 中的 p_a_colors 中删除所有列,所以我得到:
p_r_colors=np.array([[0,0,0],
[0,2,0],
[3,2,4]])
我认为,应该像
p_r_colors= np.delete(p_a_colors, np.where(np.all(p_a_colors==p_rem, axis=0)),0)
但我只是没有得到正确的轴或 [:]。
我知道
p_r_colors=copy.deepcopy(p_a_colors)
for i in range(len(p_rem)):
p_r_colors= np.delete(p_r_colors, np.where(np.all(p_r_colors==p_rem[i], axis=-1)),0)
会工作,但我试图避免(python)循环,因为我也希望性能正确。