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]
思路:使用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]