删去numpy array 中的全零行或列

最新推荐文章于 2024-10-09 13:05:24 发布

yzl14

最新推荐文章于 2024-10-09 13:05:24 发布

阅读量1.2w

点赞数

分类专栏： python初步

本文链接：https://blog.csdn.net/yzl14/article/details/100557353

版权

python初步专栏收录该内容

23 篇文章 1 订阅

订阅专栏

首先求出全零行、列的index，放在index_list中。本例中全零行和全零列index相同，故index_list相同。由index_list构建一个一维的、长度等于array的行数或列数的bool array，使得要保留的行/列所对应的元素为True，要去掉的元素为False。之后通过如下代码删去不要的行或列。注意行列不能同时删除。

# Used to filter out the all-zero rows and lines
id_list = list(set(id_list))
id_list.sort()
index_array = np.zeros(1000, dtype = bool)
for ID in id_list:
    index_array[ID] = True

synergy_array = synergy_array[index_array]
synergy_array = synergy_array[:, index_array]