使用matplotlib将非正方形数据数组保存为无空格的图形

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])

fig = plt.figure(figsize=(8, 5))  # 设置图形大小为8x5

plt.imshow(data, cmap='viridis')  # 使用viridis颜色映射

plt.axis('off')

plt.savefig('output.png', bbox_inches='tight', pad_inches=0)

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])

fig = plt.figure(figsize=(8, 5))
plt.imshow(data, cmap='viridis')
plt.axis('off')
plt.savefig('output.png', bbox_inches='tight', pad_inches=0)