knnvc_utils.py

import numpy as np

def generate_matrix_from_index(A, len=25):
    matrix = np.zeros(len, dtype=float)
    matrix[A] = 1
    return matrix


def retrieve_index_from_matrix(matrix):
    A = np.where(matrix == 1)[0]
    return A

if __name__ == '__main__':
    # Generating a matrix from index A
    A = 6
    matrix = generate_matrix_from_index(A)
    print("Generated Matrix:")
    print(matrix)

    # Retrieving index A from the matrix
    retrieved_A = retrieve_index_from_matrix(matrix)
    print("Retrieved Index A:")
    print(retrieved_A)