python绘制各区域平均房价柱状图,区域由独热编码组成
时间: 2023-07-08 11:32:33 浏览: 124
python 画条形图(柱状图)实例
可以使用Python中的matplotlib库来绘制各区域平均房价的柱状图。首先需要将独热编码转换为区域名称,可以使用pandas库中的get_dummies函数来实现。假设独热编码存储在一个名为encoded_data的DataFrame中,代码如下:
```python
import pandas as pd
# 转换独热编码为区域名称
area_names = encoded_data.columns.values
area_names = [name.split('_')[1] for name in area_names]
# 计算各区域平均房价
avg_prices = encoded_data.mean()
# 创建柱状图
import matplotlib.pyplot as plt
plt.bar(area_names, avg_prices)
plt.xlabel('Area')
plt.ylabel('Average Price')
plt.title('Average House Price by Area')
plt.show()
```
这段代码首先通过get_dummies函数将独热编码转换为DataFrame,然后使用split函数提取区域名称。接下来计算各区域平均房价,并使用matplotlib库中的bar函数创建柱状图。最后,添加标签和标题,并调用show函数显示图表。
阅读全文