개발/Python

[파이썬] Matplotlib에서 그리드 간격 변경 및 눈금 레이블 지정

MinorMan 2022. 10. 10. 04:35
반응형

<질문>

그리드 플롯에 카운트를 표시하려고 하는데 어떻게 해야 할지 모르겠습니다.

원해요:

  1. 5 간격으로 점선 그리드를 갖습니다.

  2. 20개마다 주요 눈금 레이블만 사용

  3. 틱이 플롯 외부에 있도록 하려면; 그리고

  4. 그리드 내부에 "카운트"를 갖습니다.

다음과 같은 잠재적인 중복 항목을 확인했습니다.here 그리고here, 그러나 그것을 알아낼 수 없었습니다.

이것은 내 코드입니다.

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 a "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 the minor grid to be 5 and the major grid to be 20.
plt.grid()

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

이것이 내가 얻는 것입니다.

이것이 내가 얻은 것입니다:

데이터 포인트를 덮어쓰는 문제도 있습니다.

아무도이 문제를 도와 주시겠습니까?


<답변1>

코드에 몇 가지 문제가 있습니다.

먼저 큰 것:

  1. 루프를 반복할 때마다 새 그림과 새 축을 만들고 있습니다. → 루프 외부에 fig = plt.Figureax = 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()

출력은 다음과 같습니다.

result


<답변2>

미묘한 대안MaxNoe's answer 여기서 틱을 명시적으로 설정하지 않고 대신 케이던스를 설정합니다.

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 Custom Grid

반응형