У меня есть точечная диаграмма следующего формата:
Каждая точка на этой диаграмме представляет галактику со свойством, известным как скорость звездообразования. Я хочу задать цветовую карту, зависящую от этой переменной, т. е. при более высоких значениях SFR цвет становится более синим, а при более низких значениях SFR цвет становится более красным. (В этом упражнении игнорируйте часть гистограммы с группировкой.)
Как бы я этого добился? Нужно ли мне создавать и изменять свой собственный cmap?
вам просто нужен точечный график точек с координатами x и y и цветом, представляющим третью переменную?
для этого нужен разброс, просто используйте:
import matplotlib.pyplot as plt
plt.scatter(x, y, c=z, cmap='jet')
вы можете указать любую другую цветовую карту, все возможности показаны здесь: http://matplotlib.org/examples/color/colormaps_reference.html
вот небольшой пример:
import matplotlib.pyplot as plt
import numpy as np
x = numpy.random.normal(0, 2, 100)
y = numpy.random.normal(0, 2, 100)
r = np.sqrt(x**2 + y**2)
plt.scatter(x, y, c=r, cmap='jet')
это даст вам 100 2d-гауссовых распределенных точек с цветами в зависимости от расстояния до (0,0)
