python绘制甘特图

甘特图.png

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.dates as mdates

# 准备数据
data = {
    '任务名称': ['开发计划', '可行性分析报告', '业务分析与建模', '需求建模和用例分析',
                 '架构设计和类设计', '数据库设计和物理视图设计', '总结'],
    '开始时间（天）': [1, 2, 3, 4, 5, 6, 7],
    '结束时间（天）': [2, 3, 4, 5, 6, 7, 8],
    '持续天数': [1, 1, 1, 1, 1, 1, 1],
    '责任人': ['夏浩祥'] * 7
}

df = pd.DataFrame(data)

# 将开始时间和结束时间转换为日期时间格式（以天为单位模拟日期，统一使用一个原点）
origin_date = pd.Timestamp('2024-12-16')
for col in ['开始时间（天）', '结束时间（天）']:
    try:
        df[col] = pd.to_datetime(df[col], unit='D', origin=origin_date)
    except ValueError as e:
        print(f"日期转换出现问题，列 {col} 的数据格式可能有误，错误信息：{e}")

# 创建图形和坐标轴对象
fig, ax = plt.subplots(figsize=(10, 8))

# 设置中文字体为黑体，确保能正常显示中文
plt.rcParams['font.sans-serif'] = ['SimHei']
# 解决负号显示问题（有时候可能会出现负号显示异常，一并处理）
plt.rcParams['axes.unicode_minus'] = False

# 绘制甘特图，使用水平的条形图来表示任务时间区间
for index, row in df.iterrows():
    bar_width = (row['结束时间（天）'] - row['开始时间（天）']).days
    bar = ax.barh(row['任务名称'], width=bar_width, left=row['开始时间（天）'], height=0.8)
    # 计算文本注释在任务条内相对靠右的位置（这里简单取任务条宽度的80%位置处）
    text_x = row['开始时间（天）'] + pd.Timedelta(days=bar_width * 0.8)
    ax.text(text_x, row['任务名称'], row['责任人'], ha='left', va='center')

# 设置坐标轴格式
ax.xaxis_date()
ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d'))
ax.set_ylabel('任务名称')
ax.set_xlabel('时间')
ax.set_title('论坛管理系统开发甘特图')

# 自动调整子图布局，避免标签等显示不全
plt.tight_layout()

# 显示图形
plt.show()