如何按索引屏蔽 numpy 二维数组？

Question

我有一个范围为 800 的 numpy 二维数组。现在我只想使用索引范围为 600 到 700 的元素，并将所有其他值设置为 np.nan。 持久元素的索引应该和以前一样。 数组的长度也应该相同。 谢谢！

Answer 1

import numpy as np
a = np.random.random ([800,4])

a[:600, :], a[700:, :] = np.nan, np.nan