1 2 3 4 5 6 7 8 9 10 11 12
|
# plotting a 3D scatter plot
from mpl_toolkits.mplot3d import Axes3D
fig = plt.figure()
ax = fig.add_subplot(111, projection = '3d')
classes = (1,5,7,9,11,15)
ax.scatter(data_train['B1'], data_train['B2'], data_train['B3'], c = 'r', marker = 'o', s=100, label = '1')
ax.scatter(data_train['B1'], data_train['B2'], data_train['B3'], c = 'm', marker = 'o', s=100, label = '5')
ax.scatter(data_train['B1'], data_train['B2'], data_train['B3'], c = 'c', marker = 'o', s=100, label = '7')
ax.scatter(data_train['B1'], data_train['B2'], data_train['B3'], c = 'm', marker = 'o', s=100, label = '9')
ax.scatter(data_train['B1'], data_train['B2'], data_train['B3'], c = 'k', marker = 'o', s=100, label = '11')
ax.scatter(data_train['B1'], data_train['B2'], dat |
Partager