35 lines
963 B
Python
35 lines
963 B
Python
import pandas as pd
|
|
import numpy as np
|
|
from loguru import logger
|
|
|
|
def generate(output_dir):
|
|
logger.info("开始生成进出口贸易图表")
|
|
|
|
months = ['1月', '2月', '3月', '4月', '5月', '6月']
|
|
exports = np.random.randint(2800, 3200, 6)
|
|
imports = np.random.randint(2000, 2400, 6)
|
|
|
|
data = pd.DataFrame({
|
|
'月份': months,
|
|
'出口(亿美元)': exports,
|
|
'进口(亿美元)': imports
|
|
})
|
|
data = data.set_index('月份')
|
|
|
|
from dynamic_generator import DynamicContentGenerator
|
|
generator = DynamicContentGenerator()
|
|
output_path = generator.generate_chart_matplotlib(
|
|
data,
|
|
title='进出口贸易情况',
|
|
x_label='月份',
|
|
y_label='金额 (亿美元)',
|
|
filename='trade_chart.png',
|
|
kind='bar'
|
|
)
|
|
|
|
return output_path
|
|
|
|
if __name__ == "__main__":
|
|
from pathlib import Path
|
|
generate(Path.cwd())
|