Я пытаюсь построить два numpy array, используя matplotlib, но я не получаю правильное расположение фигур. Ниже кода, который я использую:
import matplotlib.pyplot as plt
data.shape
(2, 766, 1398) # values between -1/+1
f = plt.figure(figsize=(15,15))
f.add_subplot(1,2,1)
plt.imshow(data[0], cmap="RdYlGn")
plt.colorbar()
f.add_subplot(1,2,2)
plt.imshow(data[1], cmap="RdYlGn")
plt.colorbar()
Вывод, который я получаю (в Jupyter Notebook), выглядит следующим образом:
Можно ли связать размер бара с размером изображения? Я проверил некоторые онлайн-уроки и, используя тот же код, они получают правильный вывод. Чего мне не хватает?
Кроме того, мой массив data может иметь форму (3, 766, 1398), (4, 766, 1398), (5, 766, 1398) и т. Д. Поэтому я хотел бы автоматически настроить размер подзаговора дляРазмеры моего массива. Так что, если у меня есть (2, 766, 1398), 1 строка и 2 столбца должно быть хорошо. Если 3, 2 строки - 2 столбца;если 4 2 строки - 2 столбца, если 5 3 строки - столбцы и т. д.
Возможно ли это с matplotlib?