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

Cách bỏ qua ngày cuối tuần (ngày trống) trong biểu đồ tài chính Matplotlib Python

Để bỏ qua ngày cuối tuần trong biểu đồ tài chính, ta lặp qua thời gian trong DataFrame và không vẽ nếu weekday là 5 hoặc 6 (Thứ 7, Chủ Nhật).

Các bước thực hiện

  • Thiết lập kích thước hình và khoảng đệm.
  • Tạo DataFrame với khóa time.
  • Lặp zip index và time của DataFrame.
  • Nếu timestamp có weekday 5 hoặc 6 thì bỏ qua (pass).
  • Ngược lại, vẽ điểm.
  • Đặt tick locations cho trục Y.
  • Bật lưới grid(True).
  • Hiển thị bằng show().

Ví dụ

import pandas as pd
from matplotlib import pyplot as plt

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

df = pd.DataFrame(dict(time=list(pd.date_range(start="2021-01-01", end="2021-01-15"))))

for i, t in zip(df.index, df.time):
    if t.weekday() in (5, 6):
        pass
    else:
        plt.plot(i, t, marker="*", ms=10)

plt.yticks(df.time)
plt.grid(True)
plt.show()

Kết quả

Cách bỏ qua ngày cuối tuần (ngày trống) trong biểu đồ tài chính Matplotlib Python