文本热力图Python代码
时间: 2023-11-19 16:09:48 浏览: 103
文本热力图可以使用Python中的matplotlib库来实现。下面是一个简单的例子:
```python
import matplotlib.pyplot as plt
import numpy as np
# 定义文本
text = "Hello world! This is a text heatmap example."
# 统计每个字符出现的次数
char_dict = {}
for char in text:
if char not in char_dict:
char_dict[char] = 1
else:
char_dict[char] += 1
# 将字符和次数转化为数组
char_list = list(char_dict.keys())
char_count = np.array(list(char_dict.values()))
# 绘制热力图
fig, ax = plt.subplots()
im = ax.imshow([char_count], cmap='Reds')
# 添加坐标轴
ax.set_xticks(np.arange(len(char_list)))
ax.set_yticks([0])
ax.set_xticklabels(char_list)
ax.set_yticklabels([''])
# 添加文本
for i in range(len(char_list)):
for j in range(1):
text = ax.text(i, j, char_count[i],
ha="center", va="center", color="black")
# 添加标题
ax.set_title("Text Heatmap")
# 显示图像
plt.show()
```
运行以上代码,即可得到文本热力图。你可以修改text变量中的文本内容,生成不同的热力图。
阅读全文