Python：Select 矩陣中每列非零的最后一個值

Question

我正在尋找一個命令來選擇我的矩陣的最后一個非零元素。 不是值的索引，而是實際值。 這是一個 (1001,50001) 矩陣，其中每列的最后一個值不同。

它不是 dataframe，所以最好是 numpy 命令。

Answer 1

您可以在掩碼a!=0上使用np.argsort ：

a[(a!=0).argsort(axis=0)[-1], range(a.shape[1])]

Answer 2

我對此有一個建議。 用 1 替換列的所有非零值。然后找到等於 1 的最后一個元素。即使使用 Unix 使用 grep '1' 文件，這也很容易 | tail -1 等。但是，如果您不允許修改矩陣值，那么您只需提取此列並僅運行此列的腳本。 祝你好運。