在Jupyter Notebook中,你可以使用matplotlib库来绘制双Y轴的折线图。下面是绘制单价和总价双Y轴折线图的基本步骤:
导入必要的库:
import matplotlib.pyplot as plt
from mpl_toolkits.axisartist import host_subplot, GridHelper
准备数据:
# 假设有一组数据:数量(数量1,数量2,...),单价(单价1,单价2,...),总价(总价1,总价2,...)
quantities = [10, 20, 30, 40, 50]
unit_prices = [1.5, 2.0, 3.0, 2.5, 1.8]
total_prices = [unit_prices[i] * quantities[i] for i in range(len(quantities))]
创建图表并设置双Y轴:
fig, host = plt.subplots()
host.set_xlabel('Quantity') host.set_ylabel('Unit Price ($)', color='tab:blue') host.set_title('Unit Price and Total Price by Quantity')
par1 = host.twinx() par1.set_ylabel('Total Price ($)', color='tab:red')
p1, = host.plot(quantities, unit_prices, 'b-', label='Unit Price')
p2, = par1.plot(quantities, total_prices, 'r-', label='Total Price')
host.legend(loc='upper left') par1.legend(loc='upper right')
4. 显示图表:
```python
plt.show()
这段代码首先导入了matplotlib和mpl_toolkits中的host_subplot模块,然后创建了一个图表和一个辅助的轴,分别用于显示单价和总价。接着,使用plot
方法绘制了两条折线图,并通过设置label
属性来添加图例。最后,使用plt.show()
函数显示图表。
请注意,你可能需要根据你的具体数据结构和需求对上述代码进行调整。