Skip to main content
 首页 » 编程设计

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

2025年05月04日23myhome

例如,我有一个尺寸为 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]])