/Python/Python-Generating numbers according to a corellation matrix

Python-Generating numbers according to a corellation matrix

from scipy.linalg import cholesky, sqrtm
relavant_columns = ['Affecting homelife',
           'Affecting mobility',
           'Affecting social life/hobbies',
           'Affecting work',
           'Mood',
           'Pain Score',
           'Range of motion in Doc']

# df is a pandas dataframe containing the data frame from figure 1
mu = df[relavant_columns].mean().values
cov = df[relavant_columns].cov().values
number_of_sample = 10


# generate using affine transformation
#c2 = cholesky(cov).T
c2 = sqrtm(cov).T
s = np.matmul(c2, np.random.randn(c2.shape[0], number_of_sample)) + mu.reshape(-1, 1)

# transpose so each row is a sample
s = s.T

Source: stackoverflow.com

Similar codes

0 vote359 views1 answer
Use iloc to loop through dataframe
Python
0 vote331 views1 answer
Mutiple codition datafrarme
Python
7 votes359 views2 answers
Select column in pandas dataframe
Python
0 vote365 views2 answers
MultiValueDictKeyError django
Python Django
0 vote382 views1 answer
Crawl a folder python
Python
212 votes468 views6 answers
Generate random number python
Python
18 votes350 views4 answers
How to play audio in python
Python
15 votes349 views3 answers
Remove word from string python
Python
51 votes344 views6 answers
French to english
Python
35 votes345 views6 answers
Install matplotlib
Python