У меня есть следующие данные (четыре массива равной длины):
a = [1, 4, 5, 2, 8, 9, 4, 6, 1, 0, 6]
b = [4, 7, 8, 3, 0, 9, 6, 2, 3, 6, 7]
c = [9, 0, 7, 6, 5, 6, 3, 4, 1, 2, 2]
d = [La, Lb, Av, Ac, Av, By, Lh, By, Lg, Ac, Bt]
Я делаю 3D-график массивов a, b, c:
import pylab
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.scatter(a,b,c)
plt.show()
Теперь Я хочу покрасить эти разбросанные точки с помощью массива с именем 'd', чтобы: если первая буква соответствующего значения i-го элемента в d равна "L", затем окрашивайте точку красным цветом, если она начинается с цвета "A", она зеленая, и если она начинается с "B", цвет ее синий.
Итак, первая точка (1,4,9) должна быть красной, вторая (4,7,0) красная тоже, третья (5,8,7) должна быть зеленой и т.д.
Можно ли это сделать? Пожалуйста, помогите, если у вас есть идея:)