#5, First Floor, 4th Street , Dr. Subbarayan Nagar Kodambakkam, Chennai-600 024 pro@slogix.in

Office Address

  • #5, First Floor, 4th Street Dr. Subbarayan Nagar Kodambakkam, Chennai-600 024 Landmark : Samiyar Madam
  • pro@slogix.in
  • +91- 81240 01111

Social List

How to visualize the data set in 3D scatter plot using matplot library in python?
Description

To visualize the data set in 3D representation using python.

Process

  Import mpl_toolkit and related 3D objects from matplot library.

  Read the sample data.

  Initialize the figure object(add_subplot()).

  Initialize the x,y,z axis.

  Set the x-label,y-label and z-label.

Sapmle Code

#import 3D tools and objects

from mpl_toolkits.mplot3d import axes3d

#import matplot library

import matplotlib.pyplot as plt

import pandas as pd

#sample data set

data={‘salary’:[100,200,300,400,500,400,

300,200,100, 50],’age’

:[25,26,25,23,30,29,23,23,25,25],

‘rating’:[4,3.24,2.5,2.25,2,2.25,2.5,

2.75,3.2,4.2],’bonus’

:[2500,1200,900,3000,1800,1400

,850,250,750, 1000]}

#store in a data frame

df=pd.DataFrame(data)

#initialize the figure object

fig = plt.figure()

ax1 = fig.add_subplot(111, projection=’3d’)

#passing the variables in to

corresponding axis

x = df[‘salary’]

y = df[‘age’]

z = df[‘bonus’]

a = df[‘rating’]

#3D scatter plot

ax1.scatter(x, y, z, c=’g’, marker=’o’)

ax1.scatter(x, y, a, c=’r’, marker=’o’)

ax1.set_xlabel(‘x axis’)

ax1.set_ylabel(‘y axis’)

ax1.set_zlabel(‘z axis’)

plt.show()

Screenshots