首页 > 解决方案 > 内存高效切片numpy数组

问题描述

我正在使用以下代码行对 7.4GB 数组进行切片:

xTrain = cp.array(db[(db[:,0]>=trainStartDate)&(db[:,0]<=trainEndDate)][:,5:])

我不明白为什么我的内存用完了,我有 25GB。这意味着在这一行中至少制作了 3 个数组副本。那是对的吗?有没有更好的切片方法?

标签: pythonnumpymemoryslice

解决方案


推荐阅读