jupyter notebook中,第一步使用numpy库和matplotlib.pyplot库读取人口数据,数据文件populations.npz文件在D盘目录下,第二步创建画布,添加子图,在子图上绘制散点图和折线图,并保存,请给出对应步骤的代码
时间: 2024-03-01 13:54:26 浏览: 149
以下是对应步骤的代码:
```python
# 第一步:使用numpy库和matplotlib.pyplot库读取人口数据
import numpy as np
import matplotlib.pyplot as plt
data = np.load('D:/populations.npz') # 读取数据文件
data = data['data'] # 获取数据
# 第二步:创建画布,添加子图,在子图上绘制散点图和折线图,并保存
fig, ax = plt.subplots(figsize=(8, 6)) # 创建画布和子图
# 绘制散点图
ax.scatter(data[:, 0], data[:, 1], color='b', label='City 1')
ax.scatter(data[:, 0], data[:, 2], color='r', label='City 2')
ax.scatter(data[:, 0], data[:, 3], color='g', label='City 3')
# 绘制折线图
ax.plot(data[:, 0], data[:, 1], color='b', linestyle='--')
ax.plot(data[:, 0], data[:, 2], color='r', linestyle='-.')
ax.plot(data[:, 0], data[:, 3], color='g', linestyle='-')
# 添加标签和图例
ax.set_title('Population Growth')
ax.set_xlabel('Year')
ax.set_ylabel('Population')
ax.legend()
# 保存图片
plt.savefig('population.png')
```
注意:代码中的数据文件路径需要根据自己的实际情况进行修改。
阅读全文