有点晚了，但是此版本也适用
k = 0（并且不会更改数组，因此不需要进行复制）。
def kth_diag_indices(a, k): rows, cols = np.diag_indices_from(a) if k < 0: return rows[-k:], cols[:k] elif k > 0: return rows[:-k], cols[k:] else: return rows, cols
有点晚了，但是此版本也适用
k = 0（并且不会更改数组，因此不需要进行复制）。
def kth_diag_indices(a, k): rows, cols = np.diag_indices_from(a) if k < 0: return rows[-k:], cols[:k] elif k > 0: return rows[:-k], cols[k:] else: return rows, cols
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
