从numpy数组中删除多列

Question

我有随机列数组，必须将其从numpy数组中删除。当我尝试下面的代码时，不会删除那么多列。有什么建议吗？

np.array([np.delete(image[row], columns[row].astype(int), axis=0) for row in range(height)])

Answer 1

我不太确定示例中的一些内容，例如image [row]和columns [row]，但是下面的示例用词来删除多个列。以示例np.delete(n,[0,2],1)为例，对于数组n，删除axis = 1的第一行（0）和第三行（2）

n = np.array([
        [2,3,4,6],
        [3,3,0,8],
        [8,4,1,0],
        [9,4,2,0]])

print(np.delete(n,[0,2],1))

输出

[[3 6]
 [3 8]
 [4 0]
 [4 0]]