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

最新推荐文章于 2023-03-26 21:25:13 发布

原创最新推荐文章于 2023-03-26 21:25:13 发布 · 1.3w 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#numpy

python 专栏收录该内容

45 篇文章

订阅专栏

本文介绍如何使用NumPy生成二维数组并进行赋值，同时详细讲解了如何根据条件删除满足特定条件的列，提供了实用的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

生成二维数组

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.]]