在numpy数组中获取a对角线以下元素的索引可以通过以下步骤实现:
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
indices = np.tril_indices(len(a), -1)
print(indices)
完整的代码示例:
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
indices = np.tril_indices(len(a), -1)
print(indices)
输出结果为:
(array([1, 2, 2]), array([0, 0, 1]))
在这个例子中,数组a的对角线以下元素的索引为(1, 0),(2, 0),(2, 1)。其中,第一个数组表示行索引,第二个数组表示列索引。
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云