NYC Data Science Academy| Blog
Bootcamps
Lifetime Job Support Available Financing Available
Bootcamps
Data Science with Machine Learning Flagship ๐Ÿ† Data Analytics Bootcamp Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lesson
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories Testimonials Alumni Directory Alumni Exclusive Study Program
Courses
View Bundled Courses
Financing Available
Bootcamp Prep Popular ๐Ÿ”ฅ Data Science Mastery Data Science Launchpad with Python View AI Courses Generative AI for Everyone New ๐ŸŽ‰ Generative AI for Finance New ๐ŸŽ‰ Generative AI for Marketing New ๐ŸŽ‰
Bundle Up
Learn More and Save More
Combination of data science courses.
View Data Science Courses
Beginner
Introductory Python
Intermediate
Data Science Python: Data Analysis and Visualization Popular ๐Ÿ”ฅ Data Science R: Data Analysis and Visualization
Advanced
Data Science Python: Machine Learning Popular ๐Ÿ”ฅ Data Science R: Machine Learning Designing and Implementing Production MLOps New ๐ŸŽ‰ Natural Language Processing for Production (NLP) New ๐ŸŽ‰
Find Inspiration
Get Course Recommendation Must Try ๐Ÿ’Ž An Ultimate Guide to Become a Data Scientist
For Companies
For Companies
Corporate Offerings Hiring Partners Candidate Portfolio Hire Our Graduates
Students Work
Students Work
All Posts Capstone Data Visualization Machine Learning Python Projects R Projects
Tutorials
About
About
About Us Accreditation Contact Us Join Us FAQ Webinars Subscription An Ultimate Guide to
Become a Data Scientist
    Login
NYC Data Science Acedemy
Bootcamps
Courses
Students Work
About
Bootcamps
Bootcamps
Data Science with Machine Learning Flagship
Data Analytics Bootcamp
Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lessons
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook
Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories
Testimonials
Alumni Directory
Alumni Exclusive Study Program
Courses
Bundles
financing available
View All Bundles
Bootcamp Prep
Data Science Mastery
Data Science Launchpad with Python NEW!
View AI Courses
Generative AI for Everyone
Generative AI for Finance
Generative AI for Marketing
View Data Science Courses
View All Professional Development Courses
Beginner
Introductory Python
Intermediate
Python: Data Analysis and Visualization
R: Data Analysis and Visualization
Advanced
Python: Machine Learning
R: Machine Learning
Designing and Implementing Production MLOps
Natural Language Processing for Production (NLP)
For Companies
Corporate Offerings
Hiring Partners
Candidate Portfolio
Hire Our Graduates
Students Work
All Posts
Capstone
Data Visualization
Machine Learning
Python Projects
R Projects
About
Accreditation
About Us
Contact Us
Join Us
FAQ
Webinars
Subscription
An Ultimate Guide to Become a Data Scientist
Tutorials
Data Analytics
  • Learn Pandas
  • Learn NumPy
  • Learn SciPy
  • Learn Matplotlib
Machine Learning
  • Boosting
  • Random Forest
  • Linear Regression
  • Decision Tree
  • PCA
Interview by Companies
  • JPMC
  • Google
  • Facebook
Artificial Intelligence
  • Learn Generative AI
  • Learn ChatGPT-3.5
  • Learn ChatGPT-4
  • Learn Google Bard
Coding
  • Learn Python
  • Learn SQL
  • Learn MySQL
  • Learn NoSQL
  • Learn PySpark
  • Learn PyTorch
Interview Questions
  • Python Hard
  • R Easy
  • R Hard
  • SQL Easy
  • SQL Hard
  • Python Easy
Data Science Blog > Python > Python EDA Project : Salary Prediction.

Python EDA Project : Salary Prediction.

Oreste RUKUNDO
Posted on Oct 2, 2024

Introduction:

Salary prediction is a crucial application of machine learning and statistical modeling that aims to estimate the salary of an individual based on various factors such as age, education, and experience. In addition, accurate salary predictions are valuable for job seekers, employers, and policymakers alike. The objective of this project is, therefore, to analyze the various factors that affect salary predictions. To achieve this, I explored this interesting dataset by posing two key questions:

  1. Which features are strongly associated with salary prediction?
  2. How do different features interact to affect salary?

By addressing these questions, I aim to uncover meaningful insights that can help improve the accuracy and usefulness of salary predictions in various contexts.

Data:

This dataset contains information about the salaries of employees at a company. Specifically, each row represents a different employee, and the columns include various details such as age, gender, education level, job title, years of experience, and salary, all set out across six columns.

It is important to note that this dataset is intended solely for educational use. Moreover, it was generated by large language models and not collected from actual data sources. Consequently, any commercial use of the dataset is strictly prohibited. For reference, here is a link to the dataset:  https://www.kaggle.com/datasets/rkiattisak/salary-prediction-for-beginner 

Exploratory Data  Analysis:

Which features are strongly associated with salary prediction?

Based on the data analysis, Age, Years of Experience, Education Level, and Job Title are strongly associated with salary prediction. Specifically, more years of experience and a higher education level typically translate into higher salaries. Furthermore, these factors have been clearly illustrated and explained in the graphs below, which provide a more detailed visual representation of their impact on salary predictions.

Fig.1. Salary increases with years of experience and education level

How do different features interact to affect salary?

All features are positively correlated not only between each other but also with salary.

Fig.2. More years of experience accumulated correlate with higher income

 

Fig.3a. Distribution of Salary by Education Level

Fig.3b. Distribution of Salary by Education Level

The effect of education level on earnings is reflected in Fig.3a and Fig.3b. There is a general trend for higher degrees to correlate with higher salaries. But it's interesting that there are clear exceptions to this general rule. As the violin graph shows, some individuals  with just the undergraduate degree earn more than those with higher degrees. That is likely due to differences in years of experience and area of study.

Fig.4. Correlation Between Years of Experience, Age and Salary

Generally, older workers have more experience, which typically correlates with  higher salaries. People who spend more years on a job are assumed to have acquired  greater skills and responsibilities. Thatโ€™s why older workers are often the ones found in higher positions, such as management roles, which typically come with increased pay.

 

Fig.5. Job Title vs. Salary

Fig.5. shows that higher-level roles (e.g. CEO, CTO) typically command higher salaries. This reflects the fact that roles with higher responsibilities, as indicated by the C-level rank, or in-demand skills generally offer higher salaries.

Conclusion:

Employees aiming to maximize their earning potential should focus on gaining experience, acquiring relevant skills, and considering role changes to increase earning potential. Additionally, they can take advantage of predictive models and data insights to support salary negotiations, thereby demonstrating how their experience and skills align with higher salary ranges. Furthermore, leveraging industry benchmarks and networking opportunities can also contribute to a stronger position in salary discussions.

On the other hand, employers who want to be sure to set the right salary for the job should ensure that salary structures reflect experience, job responsibilities, and market trends. Itโ€™s also important to regularly review and adjust compensation in order to remain competitive. Moreover, to retain employees over time rather than risk losing them to competitors offering better salaries, employers should create clear career paths and development opportunities that align with salary increases. In doing so, employees will better understand how they can progress in their careers without having to leave the company.

Future work:

One of the factors that can also influence salaries is the location of the job. However, I had to leave that consideration out of this analysis. Ideally, I would have conducted a deeper analysis of salary predictions in different locations and their impact on business changes. Unfortunately, there are not enough features in this dataset to analyze the impact of this trend. Nonetheless, it is definitely something worth pursuing in future work.

In particular, among the questions I would seek to answer are the following:

  • Which locations correlate with the highest salaries?
  • Which locations correlate with the lowest salaries?
  • Are the higher salaries consistent only with the increased cost of living in those areas, or are there other factors at play?

By exploring these questions, a more comprehensive understanding could be gained, leading to more informed conclusions.

Quick Links:

GitHub Repository

LinkedIn profile 

Featured Image by Freepik

About Author

Oreste RUKUNDO

A data scientist with a background in Mathematics and Statistics. Typically, I have a strong foundation in quantitative analysis, problem-solving, and statistical modeling. Skilled in using mathematical techniques and statistical methods to analyze complex datasets and extract meaningful...
View all posts by Oreste RUKUNDO >

Leave a Comment

No comments found.

View Posts by Categories

All Posts 2399 posts
AI 7 posts
AI Agent 2 posts
AI-based hotel recommendation 1 posts
AIForGood 1 posts
Alumni 60 posts
Animated Maps 1 posts
APIs 41 posts
Artificial Intelligence 2 posts
Artificial Intelligence 2 posts
AWS 13 posts
Banking 1 posts
Big Data 50 posts
Branch Analysis 1 posts
Capstone 206 posts
Career Education 7 posts
CLIP 1 posts
Community 72 posts
Congestion Zone 1 posts
Content Recommendation 1 posts
Cosine SImilarity 1 posts
Data Analysis 5 posts
Data Engineering 1 posts
Data Engineering 3 posts
Data Science 7 posts
Data Science News and Sharing 73 posts
Data Visualization 324 posts
Events 5 posts
Featured 37 posts
Function calling 1 posts
FutureTech 1 posts
Generative AI 5 posts
Hadoop 13 posts
Image Classification 1 posts
Innovation 2 posts
Kmeans Cluster 1 posts
LLM 6 posts
Machine Learning 364 posts
Marketing 1 posts
Meetup 144 posts
MLOPs 1 posts
Model Deployment 1 posts
Nagamas69 1 posts
NLP 1 posts
OpenAI 5 posts
OpenNYC Data 1 posts
pySpark 1 posts
Python 16 posts
Python 458 posts
Python data analysis 4 posts
Python Shiny 2 posts
R 404 posts
R Data Analysis 1 posts
R Shiny 560 posts
R Visualization 445 posts
RAG 1 posts
RoBERTa 1 posts
semantic rearch 2 posts
Spark 17 posts
SQL 1 posts
Streamlit 2 posts
Student Works 1687 posts
Tableau 12 posts
TensorFlow 3 posts
Traffic 1 posts
User Preference Modeling 1 posts
Vector database 2 posts
Web Scraping 483 posts
wukong138 1 posts

Our Recent Popular Posts

AI 4 AI: ChatGPT Unifies My Blog Posts
by Vinod Chugani
Dec 18, 2022
Meet Your Machine Learning Mentors: Kyle Gallatin
by Vivian Zhang
Nov 4, 2020
NICU Admissions and CCHD: Predicting Based on Data Analysis
by Paul Lee, Aron Berke, Bee Kim, Bettina Meier and Ira Villar
Jan 7, 2020

View Posts by Tags

#python #trainwithnycdsa 2019 2020 Revenue 3-points agriculture air quality airbnb airline alcohol Alex Baransky algorithm alumni Alumni Interview Alumni Reviews Alumni Spotlight alumni story Alumnus ames dataset ames housing dataset apartment rent API Application artist aws bank loans beautiful soup Best Bootcamp Best Data Science 2019 Best Data Science Bootcamp Best Data Science Bootcamp 2020 Best Ranked Big Data Book Launch Book-Signing bootcamp Bootcamp Alumni Bootcamp Prep boston safety Bundles cake recipe California Cancer Research capstone car price Career Career Day ChatGPT citibike classic cars classpass clustering Coding Course Demo Course Report covid 19 credit credit card crime frequency crops D3.js data data analysis Data Analyst data analytics data for tripadvisor reviews data science Data Science Academy Data Science Bootcamp Data science jobs Data Science Reviews Data Scientist Data Scientist Jobs data visualization database Deep Learning Demo Day Discount disney dplyr drug data e-commerce economy employee employee burnout employer networking environment feature engineering Finance Financial Data Science fitness studio Flask flight delay football gbm Get Hired ggplot2 googleVis H20 Hadoop hallmark holiday movie happiness healthcare frauds higgs boson Hiring hiring partner events Hiring Partners hotels housing housing data housing predictions housing price hy-vee Income industry Industry Experts Injuries Instructor Blog Instructor Interview insurance italki Job Job Placement Jobs Jon Krohn JP Morgan Chase Kaggle Kickstarter las vegas airport lasso regression Lead Data Scienctist Lead Data Scientist leaflet league linear regression Logistic Regression machine learning Maps market matplotlib Medical Research Meet the team meetup methal health miami beach movie music Napoli NBA netflix Networking neural network Neural networks New Courses NHL nlp NYC NYC Data Science nyc data science academy NYC Open Data nyc property NYCDSA NYCDSA Alumni Online Online Bootcamp Online Training Open Data painter pandas Part-time performance phoenix pollutants Portfolio Development precision measurement prediction Prework Programming public safety PwC python Python Data Analysis python machine learning python scrapy python web scraping python webscraping Python Workshop R R Data Analysis R language R Programming R Shiny r studio R Visualization R Workshop R-bloggers random forest Ranking recommendation recommendation system regression Remote remote data science bootcamp Scrapy scrapy visualization seaborn seafood type Selenium sentiment analysis sentiment classification Shiny Shiny Dashboard Spark Special Special Summer Sports statistics streaming Student Interview Student Showcase SVM Switchup Tableau teachers team team performance TensorFlow Testimonial tf-idf Top Data Science Bootcamp Top manufacturing companies Transfers tweets twitter videos visualization wallstreet wallstreetbets web scraping Weekend Course What to expect whiskey whiskeyadvocate wildfire word cloud word2vec XGBoost yelp youtube trending ZORI

NYC Data Science Academy

NYC Data Science Academy teaches data science, trains companies and their employees to better profit from data, excels at big data project consulting, and connects trained Data Scientists to our industry.

NYC Data Science Academy is licensed by New York State Education Department.

Get detailed curriculum information about our
amazing bootcamp!

Please enter a valid email address
Sign up completed. Thank you!

Offerings

  • HOME
  • DATA SCIENCE BOOTCAMP
  • ONLINE DATA SCIENCE BOOTCAMP
  • Professional Development Courses
  • CORPORATE OFFERINGS
  • HIRING PARTNERS
  • About

  • About Us
  • Alumni
  • Blog
  • FAQ
  • Contact Us
  • Refund Policy
  • Join Us
  • SOCIAL MEDIA

    ยฉ 2025 NYC Data Science Academy
    All rights reserved. | Site Map
    Privacy Policy | Terms of Service
    Bootcamp Application