получить разбросанные трехмерные точки matplotlib для отображения цветов по продукту трех массивов

Question

Я новичок в python, я только начал изучать и использовать его в своей диссертации.

Я пытаюсь создать трехмерный график разброса, представляющий матрицу риска. Я использовал matplotlib для отображения местоположения точек до сих пор.

import numpy as np
import matplotlib.pyplot as plt

from mpl_toolkits.mplot3d import axes3d


fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

x = np.array([4, 3, 5, 4, 7, 7, 5, 7, 4, 4, 4, 4, 5, 5, 7, 7, 7, 5, 7, 3, 3, 7, 7, 6, 6])
y = np.array([2, 2, 3, 2, 2, 3, 2, 3, 2, 3, 2, 2, 2, 3, 2, 2, 2, 2, 3, 3, 2, 2, 2, 3, 2])
z = np.array([3, 7, 4, 4, 4, 7, 4, 3, 3, 3, 4, 4, 5, 5, 4, 4, 4, 5, 8, 4, 5, 4, 3, 7, 6])


ax.scatter(x, y, z)

ax.set_xlabel('Severity')
ax.set_ylabel('Probability')
ax.set_zlabel('Detectability')

rm3=ax.scatter(x,y,z,c=z, cmap='RdYlGn_r')

cbar=plt.colorbar(rm3)

cbar.set_label("RPN")

plt.xlim(1,10)
plt.ylim(1,10)

plt.show()

введите описание изображения здесь

Однако есть ли способ отображать цвета точек как произведение трех осей и назначать цветовую шкалу? или есть другой способ лучше отобразить график?

Спасибо.

Answer 1

попробуйте это:

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import axes3d

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
x = np.array([4, 3, 5, 4, 7, 7, 5, 7, 4, 4, 4, 4, 5, 5, 7, 7, 7, 5, 7, 3, 3, 7, 7, 6, 6])
y = np.array([2, 2, 3, 2, 2, 3, 2, 3, 2, 3, 2, 2, 2, 3, 2, 2, 2, 2, 3, 3, 2, 2, 2, 3, 2])
z = np.array([3, 7, 4, 4, 4, 7, 4, 3, 3, 3, 4, 4, 5, 5, 4, 4, 4, 5, 8, 4, 5, 4, 3, 7, 6])


ax.scatter(x, y, z)

ax.set_xlabel('Severity')
ax.set_ylabel('Probability')
ax.set_zlabel('Detectability')
c=x*y*z

rm3=ax.scatter(x,y,z,c=c, cmap='RdYlGn_r')

cbar=plt.colorbar(rm3)

cbar.set_label("RPN")

plt.xlim(1,10)
plt.ylim(1,10)

plt.show()