如何用 Python 視覺化資料？

Python 程式面試題實作練習

Kirill Smirnov

Data Science Consultant, Altran

matplotlib

import matplotlib.pyplot as plt

散佈圖
直方圖
盒鬚圖

資料集

import pandas as pd

diabetes = pd.read_csv('diabetes.csv')
diabetes = diabetes[[
    'n pregnant', 'plasma glucose', 'blood pressure', 'skin thickness',
    'serum insulin', 'bmi', 'age', 'test result']]
print(diabetes.head())

   n pregnant  plasma glucose  blood pressure  skin thickness  serum insulin   bmi  age test result
0           6           148.0            72.0            35.0            NaN  33.6   50    positive
1           1            85.0            66.0            29.0            NaN  26.6   31    negative
2           8           183.0            64.0             NaN            NaN  23.3   32    positive
3           1            89.0            66.0            23.0           94.0  28.1   21    negative
4           0           137.0            40.0            35.0          168.0  43.1   33    positive

散佈圖

建立散佈圖

import matplotlib.pyplot as plt

plt.scatter(
    diabetes['serum insulin'],
    diabetes['plasma glucose']
)
plt.show()

散佈圖

建立散佈圖

import matplotlib.pyplot as plt

plt.scatter(
    diabetes['serum insulin'],
    diabetes['plasma glucose']
)

plt.title('Plasma Glucose vs Serum Insulin')

plt.show()

散佈圖

建立散佈圖

import matplotlib.pyplot as plt

plt.scatter(
    diabetes['serum insulin'],
    diabetes['plasma glucose']
)

plt.title('Plasma Glucose vs Serum Insulin')

plt.xlabel('Serum Insulin')
plt.ylabel('Plasma Glucose')

plt.show()

散佈圖

直方圖

建立直方圖

import matplotlib.pyplot as plt

plt.hist(diabetes['bmi'])

plt.title('Histogram of BMI index')

plt.xlabel('BMI index')
plt.ylabel('counts')

plt.show()

直方圖

建立直方圖

import matplotlib.pyplot as plt

plt.hist(diabetes['bmi'], bins=20)

plt.title('Histogram of BMI index')

plt.xlabel('BMI index')
plt.ylabel('counts')

plt.show()

直方圖

盒鬚圖

建立盒鬚圖

import seaborn as sns

sns.boxplot('test_result', 'bmi', data=diabetes)
plt.title('Boxplot of BMI index')

plt.show()

盒鬚圖

建立盒鬚圖

import seaborn as sns

sns.boxplot(
    x='test_result',
    y='bmi',
    data=diabetes
)
plt.title('Boxplot of BMI index')

plt.show()

盒鬚圖

建立盒鬚圖

import seaborn as sns

sns.boxplot(
    y='test_result',
    x='bmi',
    data=diabetes
)
plt.title('Boxplot of BMI index')

plt.show()

盒鬚圖

一起來練習吧！

Python 程式面試題實作練習