Python 去掉数组(array)中的零/过滤数据中的零/提取数据中的非零

思路:使用mask

import numpy as np

# 创建一串数
data = np.array([0,0,0,135,30,125])

# 目的是提取后面三个非零数据
mask = data!=0 #[False False False  True  True  True]
new_data = data[mask] #[135  30 125]
posted @ 2021-05-17 22:11  陈玉兰  阅读(4991)  评论(0编辑  收藏  举报