/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

12 votes307 views2 answers
How to shutdown a computer with python
Python
11 votes347 views3 answers
Python set windows environment variable
Python
13 votes373 views6 answers
Python tkinter
Python
9 votes347 views5 answers
Pandas drop unnamed columns
Python
1 vote343 views1 answer
How to add new cell in jupyter notebook
Python
9 votes344 views3 answers
2d array python3
Python
2 votes389 views3 answers
Matplotlib show plot
Python
0 vote353 views1 answer
Use of numpy matrix in tkinter python 3
Python
3 votes362 views1 answer
Frozen
Python
0 vote378 views1 answer
Pandas dataframe column based on another column
Python