Как я могу создать изображение в python с помощью матрицы? - PullRequest
Новая
       195

Как я могу создать изображение в python с помощью матрицы?

0 голосов
/ 21 июня 2020 
m = [98.75, 97, 96.66, 42, 9, 4, 1,98.33, 95, 93.33, 44, 7, 4, 1,95,94,87.5, 38.33, 5, 0, 
0,95, 93, 85, 35.55,5,0,0,95,92,83,30,3.33,0,0,95,91,80,28,1.66,0,0,95,90,75,21.25,1.66,0,0]

Из приведенного выше списка мне нужно создать матрицу 7x7 следующим образом: 

      [,1]  [,2]  [,3]  [,4]  [,5]  [,6]  [,7]
 [1,] 98.75 98.33 95.00 95.00 95.00 95.00 95.00
 [2,] 97.00 95.00 94.00 93.00 92.00 91.00 90.00
 [3,] 96.66 93.33 87.50 85.00 83.00 80.00 75.00
 [4,] 42.00 44.00 38.33 35.55 30.00 28.00 21.25
 [5,]  9.00  7.00  5.00  5.00  3.33  1.66  1.66
 [6,]  4.00  4.00  0.00  0.00  0.00  0.00  0.00
 [7,]  1.00  1.00  0.00  0.00  0.00  0.00  0.00

из матрицы необходимо сгенерировать изображение, например:

введите описание изображения здесь

Ответы [ 2 ]

3 голосов
/ 21 июня 2020

Сначала преобразуйте список в массив с помощью numpy, затем reshape, а затем используйте ваш любимый пакет для построения, например. seaborn или matplotlib: 

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

m = [98.75, 97, 96.66, 42, 9, 4, 1,98.33, 95, 93.33, 44, 7, 4, 1,95,94,87.5, 38.33, 5, 0, 
0,95, 93, 85, 35.55,5,0,0,95,92,83,30,3.33,0,0,95,91,80,28,1.66,0,0,95,90,75,21.25,1.66,0,0]


data = np.asarray(m).reshape(7,7)[::-1]
sns.heatmap(data, cmap = 'jet')
# plt.imshow(data)
plt.xlabel('Gain USRP2 (receiver side) (db)')
plt.ylabel('Gain USRP1 (sender side) (db)')

enter image description here

введите описание изображения здесь

2 голосов
/ 21 июня 2020

Мне нравится использовать plt.matshow(): 

import numpy as np
import matplotlib.pyplot as plt

m = [98.75, 97, 96.66, 42, 9, 4, 1,98.33, 95, 93.33, 
     44, 7, 4, 1,95,94,87.5, 38.33, 5, 0, 0,95, 93, 
     85, 35.55,5,0,0,95,92,83,30,3.33,0,0,95,91,80,28,
     1.66,0,0,95,90,75,21.25,1.66,0,0]

plt.matshow(np.array(m).reshape(7, 7).T)
plt.show()

введите описание изображения здесь

...