Skip to content

Commit 4a6dbe2

Browse files
committed
Added streamlit example for 03_05
1 parent 791ba6e commit 4a6dbe2

1 file changed

Lines changed: 28 additions & 0 deletions

File tree

code/chapter_03_video_05.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
import matplotlib.pyplot as plt
2+
import numpy as np
3+
import pandas as pd
4+
import seaborn as sns
5+
import streamlit as st
6+
7+
# First, we'll load our level up data.
8+
employees = pd.read_csv('../data/level_up_data.csv')
9+
10+
employee_numeric = employees.select_dtypes(include = np.number)
11+
12+
st.title('Visualizing Employee Data')
13+
14+
st.sidebar.title('Select your visualization variables')
15+
16+
# We will let users select 2 variables to visualize.
17+
viz_variables = st.sidebar.multiselect(
18+
'Select variables to visualize',
19+
employee_numeric.columns,
20+
max_selections = 2
21+
)
22+
23+
color_variables = st.sidebar.selectbox(
24+
'Select variable for color',
25+
employee_numeric.columns
26+
)
27+
28+
st.scatter_chart(employee_numeric, x = viz_variables[0], y = viz_variables[1], color=color_variables)

0 commit comments

Comments
 (0)