#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed Aug 4 18:57:16 2021
@author: haascp
"""
import altair as alt
[docs]def plot_1D_data(df, xlabel='', ylabel='', title='', color=None, reduce_data=True):
"""
Plots a set of 1D data.
"""
if reduce_data:
fac = df.shape[0] // 1000
if fac > 0:
df = df[::fac]
if color is None:
color = "black"
chart = alt.Chart(df, title=title).mark_line().encode(
x=alt.X(df.columns[0], axis=alt.Axis(title=xlabel)),
y=alt.Y(df.columns[1], axis=alt.Axis(title=ylabel)),
color=alt.value(color)
).configure_axis(
grid=False,
titleFontSize=16,
titleFontWeight='normal'
).configure_view(
strokeWidth=0
).interactive()
return chart
[docs]def plot_1D_layer(df, xlabel='', ylabel='', title='', color=None, reduce_data=True):
"""
Plots a set of 1D data.
"""
if reduce_data:
fac = df.shape[0] // 1000
if fac > 0:
df = df[::fac]
if color is None:
color = "black"
chart = alt.Chart(df, title=title).mark_line().encode(
x=alt.X(df.columns[0], axis=alt.Axis(title=xlabel)),
y=alt.Y(df.columns[1], axis=alt.Axis(title=ylabel)),
color=alt.value(color)
)
return chart
[docs]def plot_1D_scatter(df, xlabel='', ylabel='', title='', color=None, reduce_data=True):
"""
Plots a set of 1D data.
"""
if reduce_data:
fac = df.shape[0] // 1000
if fac > 0:
df = df[::fac]
if color is None:
color = "black"
chart = alt.Chart(df, title=title).mark_circle(size=60).encode(
x=alt.X(df.columns[0], axis=alt.Axis(title=xlabel, tickMinStep=1)),
y=alt.Y(df.columns[1], axis=alt.Axis(title=ylabel),
scale=alt.Scale(zero=False)),
tooltip=[df.columns[0], df.columns[1]],
color=alt.value(color)
).configure_axis(
grid=False,
titleFontSize=16,
titleFontWeight='normal'
).configure_view(
strokeWidth=0
).interactive()
return chart
[docs]def plot_1D_scatter_layer(df, xlabel='', ylabel='', title='', color=None,
reduce_data=True):
"""
Plots a set of 1D data.
"""
if reduce_data:
fac = df.shape[0] // 1000
if fac > 0:
df = df[::fac]
chart = alt.Chart(df, title=title).mark_circle(size=60).encode(
x=alt.X(df.columns[0], axis=alt.Axis(title=xlabel, tickMinStep=1)),
y=alt.Y(df.columns[1], axis=alt.Axis(title=ylabel)),
tooltip=[df.columns[0], df.columns[1]],
color=alt.value(color)
)
return chart