使用pyecharts写一个简单的人口迁移图
时间: 2024-04-30 22:25:04 浏览: 96
echarts实现的迁徙图示例
好的,以下是一个简单的人口迁移图的代码,使用了 pyecharts 库:
```python
from pyecharts.charts import Geo
from pyecharts import options as opts
# 定义城市名称及人口数据
cities = ['北京', '上海', '广州', '深圳', '杭州', '南京', '成都', '重庆']
populations = [2171.0, 2424.8, 1499.0, 1303.5, 981.3, 846.8, 1601.0, 3048.8]
# 定义迁出城市及迁入城市
data = [('北京', '上海'), ('上海', '广州'), ('广州', '深圳'), ('深圳', '杭州'),
('杭州', '南京'), ('南京', '成都'), ('成都', '重庆')]
# 根据城市名称及人口数据生成地图
geo = Geo().add_schema(maptype='china').set_series_opts(label_opts=opts.LabelOpts(is_show=False))
for i in range(len(cities)):
geo.add_coordinate(cities[i], i + 1, populations[i])
geo.add('', [(city, population) for city, population in zip(cities, populations)], type_='effectScatter')
# 根据迁徙数据生成迁徙图
for city_from, city_to in data:
geo.add('', [(city_from, 1), (city_to, 1)], type_='lines', effect_opts=opts.EffectOpts(symbol='arrow'))
# 设置全局配置项
geo.set_global_opts(title_opts=opts.TitleOpts(title='人口迁移图'), visualmap_opts=opts.VisualMapOpts(max_=3500))
# 输出HTML文件
geo.render('population_migration.html')
```
这个代码会生成一个人口迁移图,并输出到名为 `population_migration.html` 的 HTML 文件中。可以使用浏览器打开该文件查看效果。
阅读全文