Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách tự động chú thích giá trị lớn nhất trong Pyplot?


Để chú thích giá trị lớn nhất trong Pyplot, chúng ta có thể thực hiện các bước sau:

  • Thiết lập kích thước figure và điều chỉnh padding giữa và xung quanh subplots.
  • Tạo figure mới hoặc kích hoạt figure hiện có.
  • Tạo danh sách điểm dữ liệu x và y.
  • Vẽ x và y bằng numpy.
  • Tìm max trong mảng Y và vị trí tương ứng với phần tử max đó.
  • Chú thích điểm đó với "local max".
  • Để hiển thị figure, dùng phương thức show().

Ví dụ

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

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

x = np.array([1, 3, 5, 3, 1])
y = np.array([2, 1, 3, 1, 2])

line, = ax.plot(x, y)
ymax = max(y)
xpos = np.where(y == ymax)
xmax = x[xpos]

ax.annotate('local max', xy=(xmax, ymax), xytext=(xmax, ymax + 5), arrowprops=dict(facecolor='black'))

plt.show()

Kết quả

Cách tự động chú thích giá trị lớn nhất trong Pyplot?