繁体   English   中英

删除numpy.ndarray的最后N个元素后的内存

[英]Memory after removing last N elements of numpy.ndarray

我有一个巨大的numpy.ndarray图像array1 ,加载到RAM时需要60GB。 我需要删除该数组的最后n元素。 一个简单的解决方案是:

array1 = array1[:n-1]

但是,当我这样做时,我没有在RAM中获得任何空间,为什么呢? 由于删除了这些元素,如何获得RAM中的可用空间? 我最初是为了获得收益而进行移除。

array1[:n-1]是一个视图,一个新的数组,与原始array1共享数据缓冲区。 即使您重新分配array1 ,其数据缓冲区也不会调整大小。

array1.resize(n-1) -文档表明已调整大小/重新分配了数据缓冲区,前提是该缓冲区不与其他任何共享。

In [1105]: arr=np.arange(1000)
In [1106]: arr.nbytes
Out[1106]: 4000
In [1107]: sys.getsizeof(arr)   # those bytes plus overhead
Out[1107]: 4048
In [1108]: arr = arr[:500]      # your slice
In [1109]: arr.nbytes           # fewer bytes
Out[1109]: 2000
In [1110]: sys.getsizeof(arr)   # just the overhead
Out[1110]: 48

sys.getsizeof获取视图的大小,但是由于它与原始arr共享缓冲区,因此我们仅看到“开销”。 原始的arr仍然存在,但无法通过名称访问。

In [1111]: arr=np.arange(1000)
In [1112]: arr.resize(500)
In [1113]: arr.nbytes
Out[1113]: 2000
In [1114]: sys.getsizeof(arr)
Out[1114]: 2048

使用resize方法,似乎已调整了数据缓冲区的大小,从而释放了一半的空间。 但是我不确定是否有一种很好的测试方法,至少对于像这样的小型阵列来说不是。

潜在地,我们有3个管理内存的系统-numpy,python解释器和系统。 我们必须进一步深入研究代码(可能是C-api),以找出在resize之后是将内存添加到某种numpy缓存中,还是由Python垃圾收集器收集或返回给系统。

============

resize然后重新resize shape似乎可以减小沿第一个轴的尺寸:

In [1120]: arr = np.arange(100).reshape(10,10).copy()
In [1121]: arr.resize(50)
In [1122]: sys.getsizeof(arr)
Out[1122]: 248
In [1123]: arr = np.arange(100).reshape(10,10).copy()
In [1124]: sys.getsizeof(arr)
Out[1124]: 456
In [1125]: arr.resize(50)
In [1126]: sys.getsizeof(arr)
Out[1126]: 248
In [1127]: arr.shape
Out[1127]: (50,)
In [1128]: arr.shape=(5,10)   # inplace reshape
In [1129]: arr
Out[1129]: 
array([[ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14, 15, 16, 17, 18, 19],
       [20, 21, 22, 23, 24, 25, 26, 27, 28, 29],
       [30, 31, 32, 33, 34, 35, 36, 37, 38, 39],
       [40, 41, 42, 43, 44, 45, 46, 47, 48, 49]])

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM