Как случайным образом задать только «n» (скажем, 3) цвета для точек диаграммы рассеяния в python?
Сгенерированные случайные числа x, y, z, затем поместите их в трехмерный график разброса, но я не могу построить точки случайным образом с 3 определенными цветами (скажем, красным, черным, желтым).
В документации matplotlib.pyplot.scatter я не могу понять остальные 3 способа указания цветов, кроме 1-го.
Код:
import pandas as pd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d, Axes3D
x, y, z = np.random.rand(3,50)
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(x,y,z,marker='.',color='b')
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
plt.show()
Ответы 1
Если вы просто хотите случайным образом назначить цвет каждой точке из набора n цветов, вы можете сделать это следующим образом:
import pandas as pd
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d, Axes3D
x, y, z = np.random.rand(3,50)
n=3
colors = np.random.randint(n, size=x.shape[0])
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(x,y,z,marker='.',c=colors)
ax.set_xlabel('X Label')
ax.set_ylabel('Y Label')
ax.set_zlabel('Z Label')
plt.show()
и если вы хотите указать цветовую палитру, замените определение массива colors на: colors = np.random.choice(['red','black','yellow'], size=x.shape[0])
спасибо что работает! не могу принять до 8 минут!
что касается вашего вопроса о документации аргумента c: 1-й вариант - всего один цвет, например. 'red' - все точки будут красными. 2-й пункт - список цветов, например. ['красный', 'синий', 'черный', 'красный', ...] - один цвет на точку. Количество точек данных должно соответствовать длине массива c (этот подход используется в команде, приведенной в комментарии). 3-й пункт - вы передаете числа, которые отображаются в масштабе (от 0 до 1, я полагаю) - matplotlib имеет несколько цветовых карт, которые просто отображают число в цвет (например, от черного к белому, от синего к красному через фиолетовый и т. д.). 4-й пункт - вы даете значения RGB, по 3 на каждый пункт
не могли бы вы уточнить 3 пункта документации: цвет маркера. Возможные значения: строка формата одного цвета. Последовательность спецификаций цвета длины n. Последовательность из n чисел, которые нужно преобразовать в цвета с помощью cmap и norm. Двумерный массив, в котором строки имеют формат RGB или RGBA.