Насколько я посмотрел, я не могу найти, как построить элементы списка с цветом, соответствующим числу в той же позиции индекса в другом списке.
Что я хочу, например:
список комплексных чисел (2D-координаты)
[(-0.9643939393939394+0.1477272727272727j),
(-0.9643939393939394+0.17045454545454541j),
(-0.9643939393939394+0.19318181818181812j),
(-0.9643939393939394+0.21590909090909105j),
(-0.9643939393939394+0.23863636363636376j),
(-0.9378787878787878-0.23863636363636365j)]
список соответствующих им чисел
[4, 5, 5, 5, 6, 6]
Таким образом, первая комплексная координата - это цвет, соответствующий первому числу в другом списке при построении графика.
Я хотел бы сделать это с большим диапазоном чисел в списке, чтобы создать градиент, такой как этот:
fig, ax = plt.subplots()
v = np.array([1,2,3,4,5,6,7,8,9,10])
v = np.expand_dims(v, axis=1)
ax.imshow(v, cmap="plasma")
Таким образом, каждая координата имеет цвет, связанный с ней числом в другом списке.
Кроме того, я бы хотел, чтобы каждая координата, имеющая соответствующее число выше определенного порога, была черной. ie. в этом случае координата, связанная с числом выше 10, должна быть черной.
- Для конкретного примера: через сгенерированный градиент у меня должны быть вторая, третья и четвертая координаты в комплексном списке, чтобы быть коричнево-красный sh цвет при нанесении