使用numpy数组中的argmax获取最大值的索引

Question

1. 我分配了大约34G的内存并读取了以下数据：sst = np.empty（[365,3600,7200]，dtype-np.float32）

2. 使用argmax（sst，axis = 0）获取每个像素的最大值的索引。

3. 运行程序时出现内存错误问题。

任何人都可以帮我吗？有没有其他方法可以获得每个像素的最大值索引[365,3600,7200]？谢谢！

Answer 1

This source code表明argMax返回一个由该轴上的max填充的数组。您可以说为每个轴分配一个新数组，并在该数组上执行np.argMax，而不是这样做。如果确保在获得该最大值的索引后覆盖旧数组，则应该减少内存问题。

如果你正在做很多处理内存不足问题的事情，可能值得研究dask。