如何從 plt.imshow() 或 plt.matshow() 中提取數據？

Question

import numpy as np
import matplotlib.pyplot as plt 

data = np.random.random((10,10))
im = plt.imshow(data)
plt.axis('off')
plt.savefig("question2.png",bbox_inches='tight',pad_inches=0)
plt.show()

如何從 plt.imshow() 或 plt.matshow() 中提取數據？

Answer 1

要獲取您在 matplotlib 軸上繪制的圖像的 RGBA 數組，首先，您要獲取圖像對象（此處為im3 ）。 其次，獲取它的顏色圖（這里是ccmap ）。 最后一步，將數據數組im3._A傳遞給ccmap 。

import matplotlib.cm as cm
import numpy as np
import matplotlib.pyplot as plt 

data = np.random.random((10,10))
# imshow or matshow is OK
#im3 = plt.imshow(data, cmap="viridis_r")  #any colormap will do
im3 = plt.matshow(data, cmap="viridis_r")
plt.axis('off')
#plt.savefig("question2.png",bbox_inches='tight',pad_inches=0)
plt.show()

# get the colormap used by the previous imshow()
ccmap = im3.get_cmap()
print(ccmap.name)  # 'viridis_r'

# get the image data ***YOU ASK FOR THIS***
img_rgba_array = ccmap(im3._A)

# plot the image data
ax = plt.subplot(111)
ax.imshow(img_rgba_array);  #dont need any cmap to plot

示例輸出圖：