Я работаю с некоторыми сигналами и мне нужна сетка. Для данных, которые имеют x-диапазон [0,20.500]
и y-диапазон [0,1800]
, я попробовал следующий код:
%pylab
def grid(imgIn,bounds, axlabels):
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.ticker
dpi=100
size = (imgIn.size[0]/dpi,imgIn.size[1]/dpi)
print(f'size={size}')
plt.figure(figsize=size)
plt.imshow(imgIn,'gray') # ,vmin=-1,vmax=1)
plt.minorticks_on()
from matplotlib.ticker import StrMethodFormatter
plt.gca().xaxis.set_major_formatter(StrMethodFormatter('{x:,.0f}'))
plt.gca().yaxis.set_major_formatter(StrMethodFormatter('{x:,.2f}'))
import matplotlib.ticker as plticker
plt.gca().xaxis.set_minor_locator\
(matplotlib.ticker.MultipleLocator(bounds[0][1]))
plt.gca().yaxis.set_minor_locator\
(matplotlib.ticker.MultipleLocator(bounds[1][1]))
plt.grid(which="both", linewidth=0.72,color="k")
plt.tick_params(which="minor", length=0)
plt.xlabel(axlabels[0])
plt.ylabel(axlabels[1])
import io
buf = io.BytesIO()
plt.savefig(buf, format='png', dpi=dpi,bbox_inches='tight')
buf.seek(0)
plt.close()
bytes = buf.getvalue()
return bytes
from PIL import Image as PImage
import requests
from io import BytesIO
response = requests.get('https://i.pinimg.com/564x/d2/cf/81/d2cf8152167a978747d15f3ce3bd6835.jpg')
imgaa = PImage.open(BytesIO(response.content))
T = 20.5
yMax = 1800
imgBytes = grid(imgaa,[(0,T),(0,yMax)],('Elapsed Time','Frequency'))
import io
fig = plt.figure()
ax = fig.add_axes([0,0,1,1])
ax.axis('off')
im = PImage.open(io.BytesIO(imgBytes))
ax.imshow(im,aspect='auto')
Так что с положительной стороны - это создает сетку. С другой стороны:
- диапазоны сетки не соблюдаются
- ось Y отсчитывается от вершины и увеличивается вниз
Я не уверен, как исправить эти две проблемы. Указатели оценены!