python—numpy二维矩阵删除某一列

生成二维数组

word = np.zeros([2, 4])#生成一个二维四列的矩阵
word[0, 0] = 0 #赋值
word[0, 1] = 1
word[0, 2] = 2
word[0, 3] = 3
word[1, 0] = 4
word[1, 1] = 5
word[1, 2] = 6
word[1, 3] = 7
print(word)

根据条件删除一列或多列数组

初学者，不太懂这个，delete原来是临时删除，简直是坑爹，所以要是想要一次删除满足条件的列，需要记录列号。

上代码：

j=0
my_list = []
for i in range(1, word.shape[0]):
     while j in range(0, word.shape[1]):
         my_date = word[i][j]
         if my_date > 5:
             my_list.append(j)
         j += 1

my_word = np.delete(word, my_list, axis=1)
print(my_word)
print(word)

输出：

[[0. 1. 2. 3.]
 [4. 5. 6. 7.]]
[[0. 1.]
 [4. 5.]]
[[0. 1. 2. 3.]
 [4. 5. 6. 7.]]