import matplotlib.pyplot as plt import seaborn as sns import pandas as pd plt.rcParams['font.sans-serif']=['SimHei'] #设置中文显示 plt.rcParams['axes.unicode_minus']=False #解决中文显示为方块的问题 sales = pd.read_csv('a_test/car.csv', encoding='utf-8') df = pd.DataFrame(sales) pivot_df = df.pivot(index='地区', columns='品牌', values='销量') sns.heatmap(pivot_df, annot=True, fmt="d", cmap="YlGnBu") plt.title('BBA在四大城市的月销量热力图') plt.figure(figsize=(10,10)) plt.show()根据这个代码,设置热力图的大小
时间: 2023-11-29 20:03:45 浏览: 71
在这个代码中,你可以使用`plt.figure(figsize=(10,10))`来设置热力图的大小。这个函数可以在绘制图形之前调用,以设置整个图像的大小。
在你的代码中,你需要将`plt.figure(figsize=(10,10))`放在`sns.heatmap`之前,以确保设置图像大小的代码先执行:
``` python
import matplotlib.pyplot as plt
import seaborn as sns
import pandas as pd
plt.rcParams['font.sans-serif']=['SimHei'] #设置中文显示
plt.rcParams['axes.unicode_minus']=False #解决中文显示为方块的问题
sales = pd.read_csv('a_test/car.csv', encoding='utf-8')
df = pd.DataFrame(sales)
pivot_df = df.pivot(index='地区', columns='品牌', values='销量')
plt.figure(figsize=(10,10)) # 设置图像大小
sns.heatmap(pivot_df, annot=True, fmt="d", cmap="YlGnBu")
plt.title('BBA在四大城市的月销量热力图')
plt.show()
```
在上面的代码中,我们将`plt.figure(figsize=(10,10))`放在了`sns.heatmap`之前,以确保先设置图像大小。然后,我们使用`plt.show`函数来显示图像。你可以根据需要调整图像大小,以适应你的具体需求。
阅读全文