63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
|
import pandas as pd
|
||
|
|
import numpy as np
|
||
|
|
from pathlib import Path
|
||
|
|
from loguru import logger
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
|
||
|
|
def generate(output_dir):
|
||
|
|
logger.info("开始生成就业数据表")
|
||
|
|
|
||
|
|
categories = ['城镇调查失业率', '青年失业率', '制造业PMI', '服务业PMI']
|
||
|
|
current = [5.1, 15.3, 49.8, 53.2]
|
||
|
|
previous = [5.2, 15.6, 49.5, 53.0]
|
||
|
|
change = [c - p for c, p in zip(current, previous)]
|
||
|
|
|
||
|
|
data = pd.DataFrame({
|
||
|
|
'指标': categories,
|
||
|
|
'当前值': current,
|
||
|
|
'上期值': previous,
|
||
|
|
'变动': change
|
||
|
|
})
|
||
|
|
|
||
|
|
fig, ax = plt.subplots(figsize=(10, 6))
|
||
|
|
ax.axis('tight')
|
||
|
|
ax.axis('off')
|
||
|
|
|
||
|
|
table_data = []
|
||
|
|
for _, row in data.iterrows():
|
||
|
|
change_color = '#4CAF50' if row['变动'] >= 0 else '#F44336'
|
||
|
|
table_data.append([
|
||
|
|
row['指标'],
|
||
|
|
f"{row['当前值']:.1f}",
|
||
|
|
f"{row['上期值']:.1f}",
|
||
|
|
f"{row['变动']:+.1f}"
|
||
|
|
])
|
||
|
|
|
||
|
|
table = ax.table(
|
||
|
|
cellText=table_data,
|
||
|
|
colLabels=['指标', '当前值', '上期值', '变动'],
|
||
|
|
cellLoc='center',
|
||
|
|
loc='center',
|
||
|
|
colColours=['#4A90E2'] * 4
|
||
|
|
)
|
||
|
|
|
||
|
|
table.auto_set_font_size(False)
|
||
|
|
table.set_fontsize(12)
|
||
|
|
table.scale(1, 2)
|
||
|
|
|
||
|
|
for key, cell in table.get_celld().items():
|
||
|
|
if key[0] > 0 and key[1] == 3:
|
||
|
|
cell.set_text_props(color='green' if '+' in cell.get_text().get_text() else 'red')
|
||
|
|
|
||
|
|
ax.set_title('就业与PMI数据摘要', fontsize=16, fontweight='bold', pad=20)
|
||
|
|
|
||
|
|
output_path = Path(output_dir) / 'employment_table.png'
|
||
|
|
plt.savefig(str(output_path), dpi=150, bbox_inches='tight')
|
||
|
|
plt.close()
|
||
|
|
|
||
|
|
logger.info(f"就业数据表已生成: {output_path}")
|
||
|
|
return str(output_path)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
generate(Path.cwd())
|