Matplotlib 3D-графики цветов из разных классов от Dataframe

Question

Я пытаюсь построить 3D-график в Matplotlib из данных Pointcloud, которые по существу извлекаются из двух разных классов.

Однако я не могу разделить классы на разные цвета. Мой код ниже.

x=pd.DataFrame(np.array(x).reshape(-1,1))
y=pd.DataFrame( np.array(y).reshape( -1, 1 ) )
z=pd.DataFrame(np.array(z).reshape(-1,1))
target=pd.DataFrame(np.array(target).reshape(-1,1))
new_data=[x,y,z,target]

new_data = pd.concat(new_data, axis=1, ignore_index=True )

new_data.columns = ['x','y','z','target']


colors=[]

fig=plt.figure(figsize=(8,8))
ax=fig.add_subplot(111,projection='3d')
ax.scatter(new_data.x,new_data.y,new_data.z,color='target')

Аргумент цвета не может быть связан с классом в столбце «Цель» в моем фрейме данных. Я что-то упускаю?

Answer 1

Я нашел ответ сам - сопоставил Dataframe с аргументами Color, используя ниже col=new_data['target'].map({'Variable1':'r','Variable2 ':'g','Variable3':'b'})

Answer 2

вы говорите, что цвета должны исходить из значений строки 'target'. Измените его на c=new_data.target