# 处理数据 data_array = df.head(10).loc[:,['下行用户平均速率(Mbit/s)','下行PRB平均利用率(%)']].to_numpy() # print(data_array) row_names = df['下行用户平均速率(Mbit/s)'].head(10) print(row_names) col_names = df['下行PRB平均利用率(%)'].head(10) print(col_names) # 设置x,y轴坐标 plt.xlabel = row_names plt.ylabel = col_names # 设置坐标轴刻度 plt.xticks(range(0,900,100)) plt.yticks(range(0,120,20)) # 绘制热力图 sns.heatmap(data_array, cmap="GnBu", xticklabels=row_names, yticklabels=col_names) plt.rcParams['font.sans-serif'] = ['SimHei'] #解决中文乱码 plt.show()
时间: 2024-02-24 21:57:47 浏览: 122
pandas数据选取:df[] df.loc[] df.iloc[] df.ix[] df.at[] df.iat[]
这段代码的作用是将 DataFrame 中的前 10 行数据取出来,选取其中的两列数据生成一个 numpy 数组,然后使用 seaborn 库中的 heatmap 函数绘制热力图。
其中,row_names 和 col_names 分别是前 10 行数据中的两列数据的列名,用于作为热力图的 x 轴和 y 轴标签。后面的代码设置了 x 轴和 y 轴的刻度范围,以及热力图的颜色映射 cmap。
需要注意的是,设置 x 轴和 y 轴标签应该使用 xlabel 和 ylabel 函数,而不是直接赋值给 plt.xlabel 和 plt.ylabel 属性。应该将代码修改为:
```python
plt.xlabel(row_names)
plt.ylabel(col_names)
```
另外,plt.rcParams['font.sans-serif'] = ['SimHei'] 用于设置中文字符集,如果不需要显示中文,则可以省略这行代码。
阅读全文