/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

0 vote179 views1 answer
Python select columns names from dataframe
Python
3 votes179 views1 answer
Numpy array count frequency
Python
1 vote206 views1 answer
Python check if two sets intersect
Python
7 votes170 views2 answers
How to create a variablein python
Python
0 vote241 views1 answer
Pytorch multiply tensors element by elementwise
Python
8 votes137 views1 answer
How to check if string exists in array pytohn
Python
0 vote185 views1 answer
Columns overlap but no suffix specified: Index(['zpid'], dtype='object')
Python
0 vote203 views2 answers
IPTC text classification example
Python
1 vote187 views1 answer
ImportError: No module named 'transforms3d'
Python
1 vote180 views1 answer
Plt.savefig cutting off labels
Python