"""
導(dǎo)入必需的庫(kù)：
matplotlib.pyplot用于繪圖，
numpy用于數(shù)值計(jì)算，
datetime用于獲取當(dāng)前日期

"""
import matplotlib.pyplot as plt
import numpy as np
from datetime import datetime

# 出生年月日
birth_year = 1991
birth_month = 9
birth_day = 1

# 當(dāng)前年月日
current_date = datetime.now()
current_year = current_date.year
current_month = current_date.month

# 設(shè)置圖表的總年數(shù)和每行的年數(shù)
# 設(shè)置圖表的總年數(shù)為80年，每行顯示4年
total_years = 80
years_per_row = 4

# 計(jì)算總行數(shù)和總列數(shù)
total_rows = total_years // years_per_row
total_columns = years_per_row * 12

# 創(chuàng)建圖表
fig, ax = plt.subplots(figsize=(12, 8))

# 計(jì)算從出生到當(dāng)前日期已經(jīng)過(guò)去的月份數(shù)
months_passed = (current_year - birth_year) * 12 + (current_month - birth_month)

# 繪制所有月份
# for i in range(total_rows * total_columns):
#   color = 'red' if i < months_passed else 'black'
#   ax.scatter(i % total_columns, i // total_columns, marker='o', edgecolors=color, facecolors='none', s=30)


# 繪制所有月份
# 如果該月份已經(jīng)過(guò)去，用紅色邊框和綠色填充來(lái)表示，否則用黑色邊框和空心來(lái)表示。
for i in range(total_rows * total_columns):
    if i < months_passed:
        ax.scatter(i % total_columns, i // total_columns, marker='o', edgecolors='red', facecolors='green', s=150)
    else:
        ax.scatter(i % total_columns, i // total_columns, marker='o', edgecolors='black', facecolors='none', s=150)

# 添加每12列之后的分割線
"""
通過(guò)plt.axvline()函數(shù)在每12列之后添加一條分割線。
x=col - 0.5表示分割線的位置，
color='gray'設(shè)置分割線的顏色為灰色，
linestyle='--'設(shè)置分割線為虛線，
linewidth=1設(shè)置分割線的寬度為1。
這樣可以在每行顯示的4年的12個(gè)月份之間添加分割線，使圖表更清晰。
"""
for col in range(12, total_columns, 12):
    plt.axvline(x=col - 0.5, color='gray', linestyle='--', linewidth=1)

# 設(shè)置軸標(biāo)簽
ax.set_xlabel('Months')
ax.set_ylabel('Years')

# 設(shè)置軸刻度
"""

ax.set_xticks(np.arange(0, total_columns, 12))
ax.set_xticklabels(np.arange(1, years_per_row + 1))
"""

"""
ax.set_xticks(np.arange(0, total_columns + 1, 1))設(shè)置了X軸的刻度，使其每列都顯示刻度線，
而xtick_labels使用np.tile函數(shù)重復(fù)生成1到12的標(biāo)簽。
這樣可以在每個(gè)1到4的列中分別顯示1到12的刻度值。
"""
ax.set_xticks(np.arange(0, total_columns, 1))
xtick_labels = np.tile(np.arange(1, 13), 4)
# print(xtick_labels)
ax.set_xticklabels(xtick_labels)



# 設(shè)置Y軸刻度
ax.set_yticks(np.arange(0, total_rows, 1))
ax.set_yticklabels(np.arange(0, total_years, years_per_row))

# 設(shè)置標(biāo)題
ax.set_title('A 80-Year Human Life in Months')

# 隱藏右邊和上邊的軸線
ax.spines['right'].set_visible(False)
ax.spines['top'].set_visible(False)

# 顯示網(wǎng)格
# ax.grid(True)

# 反轉(zhuǎn)y軸，使得0歲在頂部
plt.gca().invert_yaxis()

# 顯示圖表
plt.tight_layout()
plt.savefig("WhellOfLife.png")
plt.show()