Укажите максимальный диапазон для осей в сетке Matplotlib - PullRequest
0 голосов
/ 24 апреля 2020

Я работаю с некоторыми сигналами и мне нужна сетка. Для данных, которые имеют 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')    

enter image description here

Так что с положительной стороны - это создает сетку. С другой стороны:

  1. диапазоны сетки не соблюдаются
  2. ось Y отсчитывается от вершины и увеличивается вниз

Я не уверен, как исправить эти две проблемы. Указатели оценены!

1 Ответ

1 голос
/ 24 апреля 2020

Я довольно много прояснил код, чтобы работать намного проще. У меня осталось следующее:

from io import BytesIO
import matplotlib.pyplot as plt
from PIL import Image as PImage
import requests

response = requests.get(
    'https://i.pinimg.com/564x/d2/cf/81/d2cf8152167a978747d15f3ce3bd6835.jpg')
image = PImage.open(BytesIO(response.content))
dpi = 100
size = (image.size[0] / dpi, image.size[1] / dpi)
fig, ax = plt.subplots(figsize=size)
ax.imshow(image, extent=[0, 20.5, 0, 1800], aspect='auto')
ax.grid(which='both', color='black', linewidth=2)
ax.minorticks_on()
ax.tick_params(which='minor', length=0)
ax.set_xlabel('Elapsed Time')
ax.set_ylabel('Frequency')
plt.show()

, которое отображает

enter image description here

Это то, что вы ищете?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...