matplotlib 以数字作为符号+图例的散点图

iswrvxsc  于 2023-06-06  发布在  其他
关注(0)|答案(1)|浏览(148)

我想画一个散点图,用彩色数字代替点作为符号。我做了如下

n=np.arange(1,14,1)

fig, axs = plt.subplots(1, 2)

axs[0].scatter(x, y, linestyle='None', color="white")

for i, txt in enumerate(n):
     axs[0].annotate(txt, (x[i], y[i]), color=x_y_colours[i], ha="center", va="center")

成功了,但现在我不知道如何创造传奇!我想有彩色数字作为符号,然后标签。

e4yzc0pl

e4yzc0pl1#

您可以使用markers in latex form,使用文本或数字作为标记。为此,您可以编写marker='$...$',类似于matplotlib标签中使用latex的方式。请注意,这些标记会自动居中。

import matplotlib.pyplot as plt
import numpy as np

n = np.arange(1, 14)
theta = np.pi * n * (3 - np.sqrt(5))
r = np.sqrt(n)
x = r * np.cos(theta)
y = r * np.sin(theta)
x_y_colours = plt.get_cmap('hsv')(n / n.max())
x_y_labels = [*'abcdefghijklm']

fig, ax = plt.subplots()
for xi, yi, color_i, label_i, txt in zip(x, y, x_y_colours, x_y_labels, n):
    ax.scatter(xi, yi, marker=f'${txt}$', s=200, color=color_i, label=label_i)
ax.legend(markerscale=0.5, bbox_to_anchor=[1.01, 1.01], loc='upper left')
plt.tight_layout()
plt.show()

相关问题