首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Numpy裁剪2D数组为非NaN值

Numpy裁剪2D数组为非NaN值
EN

Stack Overflow用户
提问于 2014-09-14 15:24:10
回答 1查看 1.2K关注 0票数 3

我有一个2D Numpy数组,其中包含一个被NaN海洋包围的数据孤岛。

我想裁剪数组,使其只包含边界框和非NaN区域的内容。

我还想知道这个框相对于原始数组的索引和大小。

这个是可能的吗?

EN

回答 1

Stack Overflow用户

发布于 2014-09-14 15:52:59

代码语言:javascript
运行
复制
import numpy as np
a = np.empty((15,10))
a.fill(np.nan)
a[7,6] = 76
a[8,5] = 85
a[9,5] = 95
a[9,7] = 97

现在我们有一个15行10列的数组,在NAN的海洋中包含一个数据孤岛:

代码语言:javascript
运行
复制
nan,  76.,  nan,
85.,  nan,  nan,
95.,  nan,  97.,

接下来:

代码语言:javascript
运行
复制
nans = np.isnan(a)
nancols = np.all(nans, axis=0) # 10 booleans, True where col is all NAN
nanrows = np.all(nans, axis=1) # 15 booleans

firstcol = nancols.argmin() # 5, the first index where not NAN
firstrow = nanrows.argmin() # 7

lastcol = len(nancols) - nancols[::-1].argmin() # 8, last index where not NAN
lastrow = len(nanrows) - nanrows[::-1].argmin() # 10

最后:

代码语言:javascript
运行
复制
a[firstrow:lastrow,firstcol:lastcol]

向我们展示小岛:

代码语言:javascript
运行
复制
array([[ nan,  76.,  nan],
       [ 85.,  nan,  nan],
       [ 95.,  nan,  97.]])
票数 8
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/25831023

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档