Измените интервал сетки и укажите метки галочки в Matplotlib

84

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

  1. Пунктирная сетка с интервалом 5

  2. Ставьте основные отметки только каждые 20

  3. Хочу, чтобы галочки были вне сюжета.

  4. Имейте "счетчики" внутри этих сеток

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

Это мой код.

import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator, FormatStrFormatter

for key, value in sorted(data.items()):
    x = value[0][2]
    y = value[0][3]
    count = value[0][4]

    fig = plt.figure()
    ax = fig.add_subplot(111)

    ax.annotate(count, xy = (x, y), size = 5)
    # Overwrites and I only get the last data point

    plt.close()
    # Without this, I get "fail to allocate bitmap" error

plt.suptitle('Number of counts', fontsize = 12)
ax.set_xlabel('x')
ax.set_ylabel('y')
plt.axes().set_aspect('equal')

plt.axis([0, 1000, 0, 1000])
# This gives an interval of 200

majorLocator   = MultipleLocator(20)
majorFormatter = FormatStrFormatter('%d')
minorLocator   = MultipleLocator(5)
# I want minor grid to be 5 and major grid to be 20
plt.grid()

filename = 'C:\Users\Owl\Desktop\Plot.png'
plt.savefig(filename, dpi = 150)
plt.close()

Вот что я получаю.

Вот что я получаю.

Еще у меня проблема с перезаписью точек данных. Может ли кто-нибудь помочь мне с этой проблемой?

сова
источник

Ответы:

194

В вашем коде есть несколько проблем.

Сначала большие:

  1. Вы создаете новую фигуру и новые оси на каждой итерации цикла → поместите fig = plt.figureи ax = fig.add_subplot(1,1,1)вне цикла.

  2. Не используйте локаторы. Вызовите функции ax.set_xticks()и ax.grid()с правильными ключевыми словами.

  3. С помощью plt.axes()вы снова создаете новые оси. Используйте ax.set_aspect('equal').

Незначительные вещи: вы не должны смешивать синтаксис, подобный MATLAB, как plt.axis()с объективным синтаксисом. Использование ax.set_xlim(a,b)иax.set_ylim(a,b)

Это должен быть рабочий минимальный пример:

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

# Major ticks every 20, minor ticks every 5
major_ticks = np.arange(0, 101, 20)
minor_ticks = np.arange(0, 101, 5)

ax.set_xticks(major_ticks)
ax.set_xticks(minor_ticks, minor=True)
ax.set_yticks(major_ticks)
ax.set_yticks(minor_ticks, minor=True)

# And a corresponding grid
ax.grid(which='both')

# Or if you want different settings for the grids:
ax.grid(which='minor', alpha=0.2)
ax.grid(which='major', alpha=0.5)

plt.show()

Выход такой:

результат

MaxNoe
источник
2
Большое спасибо за ваш ответ! Вы решили мои проблемы! Чтобы установить тики снаружи, мне просто нужно было добавить ax.tick_params (which = 'both', direction = 'out').
owl
32

Тонкая альтернатива ответу MaxNoe, где вы явно не устанавливаете тики, а вместо этого устанавливаете частоту вращения.

import matplotlib.pyplot as plt
from matplotlib.ticker import (AutoMinorLocator, MultipleLocator)

fig, ax = plt.subplots(figsize=(10, 8))

# Set axis ranges; by default this will put major ticks every 25.
ax.set_xlim(0, 200)
ax.set_ylim(0, 200)

# Change major ticks to show every 20.
ax.xaxis.set_major_locator(MultipleLocator(20))
ax.yaxis.set_major_locator(MultipleLocator(20))

# Change minor ticks to show every 5. (20/4 = 5)
ax.xaxis.set_minor_locator(AutoMinorLocator(4))
ax.yaxis.set_minor_locator(AutoMinorLocator(4))

# Turn grid on for both major and minor ticks and style minor slightly
# differently.
ax.grid(which='major', color='#CCCCCC', linestyle='--')
ax.grid(which='minor', color='#CCCCCC', linestyle=':')

Пользовательская сетка Matplotlib

getup8
источник
Благодаря! Это более практично.
M. Ka