jupyter notebook绘制城市与薪资的直方图
时间: 2024-06-11 18:03:43 浏览: 14
Jupyter Notebook是一个强大的交互式数据科学环境,常用于数据分析、可视化和机器学习。要使用它绘制城市与薪资的直方图,首先你需要有数据集包含城市的薪资信息。假设你已经有了这样的数据,并且数据已经导入到Pandas DataFrame中。以下是一般的步骤:
1. **数据准备**:
- 导入所需的库:`pandas`(用于数据处理),`matplotlib`(用于绘图)和`seaborn`(可选,增强图形美感)。
```python
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
```
2. **加载数据**:
假设你的数据存储在一个CSV文件中,使用`pd.read_csv()`读取。
```python
data = pd.read_csv('city_salaries.csv')
```
3. **数据清洗**:
确保数据无缺失值,并选择需要的城市和薪资列。
```python
data = data[['City', 'Salary']]
data.dropna(inplace=True) # 删除缺失值
```
4. **创建直方图**:
使用`plt.hist()`或`sns.histplot()`绘制直方图,其中`x`参数是薪资列,`hue`参数是城市列,用于分组。
```python
fig, ax = plt.subplots()
sns.histplot(data=data, x='Salary', hue='City', bins=20, ax=ax)
```
5. **美化图形**:
可以调整标题、轴标签、颜色等。
```python
ax.set_title('City Distribution by Salary')
ax.set_xlabel('Salary')
ax.set_ylabel('Frequency')
```
6. **显示图形**:
```python
plt.show()
```