Research breakthrough possible @S-Logix pro@slogix.in

Office Address

  • 2nd Floor, #7a, High School Road, Secretariat Colony Ambattur, Chennai-600053 (Landmark: SRM School) Tamil Nadu, India
  • 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.

Sample 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
visualize the data set in 3D scatter plot using matplot library in python