我希望能更新到上面中提到save_calendar的保存日历的脚本
时间: 2024-03-15 22:44:55 浏览: 19
好的,以下是一个更新后的 `save_calendar` 脚本,可以根据输入的年份和月份生成对应的日历图片,并在每个日期格子中添加相应的日期数字。
```python
import calendar
import matplotlib.pyplot as plt
def save_calendar(year, month, filename):
# 获取指定年份和月份的日历
cal = calendar.monthcalendar(year, month)
# 创建一个新的 figure
fig = plt.figure(figsize=(8, 6))
# 添加子图
ax = fig.add_subplot(111)
# 绘制日历
ax.matshow(cal)
# 添加日期数字
for i in range(len(cal)):
for j in range(len(cal[i])):
day = cal[i][j]
if day != 0:
ax.text(j, i, str(day), ha='center', va='center')
# 调整坐标轴和边框
ax.set_xticks([])
ax.set_yticks([])
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)
ax.spines['bottom'].set_visible(False)
ax.spines['left'].set_visible(False)
# 保存图像
plt.savefig(filename, dpi=300, bbox_inches='tight')
```
使用方法:
```python
save_calendar(2022, 1, 'calendar.png')
```
这将生成 2022 年 1 月的日历图片,并将其保存为名为 `calendar.png` 的文件。您可以根据自己的需要修改文件名和路径。