python - 在 matplotlib 中使用条件标记

Question

我有一个问题。我正在使用 stem 函数绘制图形。我的问题是在变量 values 中有很多零值。他们在我绘制的图表下画了红线。这里有没有可能在值为零时如何不做标记？谢谢。

keys = list(dictionary.keys())
values = list(dictionary.values())
plt.stem(keys,values,'-ro')
plt.show()

Answer 1

一种实现方式(不知道是不是matplotlibic方式):

• 首先过滤掉零值
• 将格式拆分为两个关键字，以阻止绘制底部圆圈
• 绘制它

import matplotlib.pyplot as plt
import random
dictionary = {i: random.choice([0] * 100 + list(range(100))) for i in range(100)}
new_dict = {i: j for i, j in dictionary.items() if j}
keys = list(new_dict.keys())
values = list(new_dict.values())

plt.stem(keys, values, markerfmt='ro', linefmt='-r')
plt.show()

结果: