例如,我有一个尺寸为 3 x 3 的二维数组。
[1 2 7
4 5 6
7 8 9]
我想删除所有包含 7 的列 - 所以第一和第三,输出一个 3 x 1 矩阵:
[2
5
8]
如何在 python 中执行此操作?我想将它应用于 n x n 维的大矩阵。
谢谢!
请您参考如下方法:
#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]])
