python - 如果列具有指定元素，如何从 numpy 数组中删除该列？ (Python)

Question

例如，我有一个尺寸为 3 x 3 的二维数组。

 [1 2 7

  4 5 6

  7 8 9]

我想删除所有包含 7 的列 - 所以第一和第三，输出一个 3 x 1 矩阵:

 [2

  5

  8]

如何在 python 中执行此操作？我想将它应用于 n x n 维的大矩阵。

谢谢!

Answer 1

#Creating array
x = np.array([[1, 2, 7],[4,5, 6],[7,8,9]])

x
Out[]: 
array([[1, 2, 7],
       [4, 5, 6],
       [7, 8, 9]])

#Deletion
a = np.delete(x,np.where(x ==7),axis=1)

a
Out[]: 
array([[2],
       [5],
       [8]])