如何用python将csv中不同价格划分高中低价格区间并绘制价格区间对需求量影响折线图代码
时间: 2023-06-03 09:04:25 浏览: 135
可以使用 pandas 和 matplotlib 库来实现。
具体步骤如下:
1. 使用 pandas 的 read_csv 方法读取 csv 文件,并将数据存储到 DataFrame 中。
2. 添加一列用于表示价格区间。
3. 利用 pivot_table 方法统计需求量,并将数据进行处理,使其能够绘制折线图。
4. 使用 matplotlib 库绘制折线图。
示例代码如下:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取 csv 文件
df = pd.read_csv('data.csv')
# 添加一列用于表示价格区间
bins = [-float('inf'), 20, 50, 100, float('inf')]
labels = ['Low', 'Medium-Low', 'Medium-High', 'High']
df['price_range'] = pd.cut(df['price'], bins=bins, labels=labels)
# 统计需求量
table = pd.pivot_table(df, index='price_range', values='demand', aggfunc='mean').reset_index()
# 绘制折线图
plt.plot(table['price_range'], table['demand'])
plt.title('Demand vs Price Range')
plt.xlabel('Price Range')
plt.ylabel('Demand')
plt.show()
```
阅读全文