def triu_indices(n, k=0):
m = numpy.ones((n, n), int)
a = triu(m, k)
return numpy.where(a != 0)
微信
公众号
扫码关注公众号
微信
公众号