Индивидуальные альфа-значения в диаграмме рассеяния/Matplotlib

Мне интересно, можно ли иметь индивидуальные альфа-значения для каждой точки, которая будет построена с использованием функции Разброс Matplotlib. Мне нужно построить набор точек, каждый со своим альфа-значением.

Например, у меня есть этот код для построения некоторых точек

def plot_singularities(points_x, p, alpha_point, file_path):
    plt.figure()
    plt.scatter(points_x, points_y, alpha=alpha_point)
    plt.savefig(file_path + '.png', dpi=100)
    plt.close()

Здесь у меня все мои points_x, points_y и alpha_point имеют n значений. Однако я не могу назначить массив параметру альфа в scater(). Мне интересно, как я могу изменить значение альфа для каждой точки.

Я знаю, что я могу сделать цикл и график по точкам с его специфическим значением альфа, но это не кажется хорошим подходом.

Спасибо.

Ответы

Ответ 1

Предложение tcaswell правильное, вы можете сделать это вот так:

import numpy as np
import matplotlib.pylab as plt

x = np.arange(10)
y = np.arange(10)

alphas = np.linspace(0.1, 1, 10)
rgba_colors = np.zeros((10,4))
# for red the first column needs to be one
rgba_colors[:,0] = 1.0
# the fourth column needs to be your alphas
rgba_colors[:, 3] = alphas

plt.scatter(x, y, color=rgba_colors)
plt.show()

Вывод