/Python/Python matrix condensed to square

Python matrix condensed to square

def square_to_condensed(i, j, n):
    assert i != j, "no diagonal elements in condensed matrix"
    if i < j:
        i, j = j, i
    return n*j - j*(j+1)//2 + i - 1 - j

Source: stackoverflow.com

Python matrix condensed to square

>>> square_to_condensed(1, 2, len(points))
3
>>> dist_condensed[3]
4.4721359549995796
>>> dist[1,2]
4.4721359549995796

Source: stackoverflow.com

Python matrix condensed to square

import math

def calc_row_idx(k, n):
    return int(math.ceil((1/2.) * (- (-8*k + 4 *n**2 -4*n - 7)**0.5 + 2*n -1) - 1))

def elem_in_i_rows(i, n):
    return i * (n - 1 - i) + (i*(i + 1))//2

def calc_col_idx(k, i, n):
    return int(n - elem_in_i_rows(i + 1, n) + k)

def condensed_to_square(k, n):
    i = calc_row_idx(k, n)
    j = calc_col_idx(k, i, n)
    return i, j

Source: stackoverflow.com

Similar codes

3 votes183 views6 answers
Python variable
Python
12 votes203 views6 answers
Print()
Python
0 vote179 views1 answer
B'string' python
Python
2 votes230 views1 answer
How do I mock an uploaded file in django?
Python Django
0 vote199 views1 answer
Tail a log file with python
Python
9 votes206 views6 answers
What is enumerate in python
Python
31 votes201 views6 answers
Count unique pandas
Python
2 votes224 views1 answer
Python add unique to list
Python
0 vote202 views1 answer
How to add condition if null value in django orm
Python Django
0 vote248 views1 answer
S = 1 + 2 + ... + n in python
Python