-
Notifications
You must be signed in to change notification settings - Fork 54
/
_lollipop.py
92 lines (76 loc) · 2.56 KB
/
_lollipop.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# -*- coding: utf-8 -*-
# _lollipop.py
# The lollipop function: plot digital samples like it was a lollipop field.
# Copyright 2013 Giuseppe Venturini
# This file is part of python-deltasigma.
#
# python-deltasigma is a 1:1 Python replacement of Richard Schreier's
# MATLAB delta sigma toolbox (aka "delsigma"), upon which it is heavily based.
# The delta sigma toolbox is (c) 2009, Richard Schreier.
#
# python-deltasigma is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# LICENSE file for the licensing terms.
"""Lollipop plotting function.
"""
from warnings import warn
import numpy as np
import pylab as plt
from numpy.matlib import repmat
# Plot lollipops (o's and sticks)
#
# ^
# | o o o o o o o o
# | | o | o o | o o | | | o | |
# | | o | | | | | | o o | | | | | | |
# +----------------------------------->
def lollipop(x, y, color=None, lw=2, ybot=0):
"""Plot lollipops (o's and sticks)
**Parameters:**
x, y : ndarray
data to be plotted
color : any matplotlib color, optional
plotting color
lw : float, optional
line width value in points
ybot : float, optional
Dummy parameter available for compatiblity
**Returns:**
None
**Example:**
.. plot::
import matplotlib.pyplot as plt
import numpy as np
from deltasigma import lollipop
t = np.arange(1, 20)*1e-3
f = 20.
a = np.sin(2*np.pi*f*t)
lollipop(t, a)
plt.gcf().set_size_inches((8, 4))
plt.grid(True)
plt.show()
"""
if ybot:
warn('lollipop() got a non-zero ybot, but only ybot=0 is ' + \
'supported. Setting ybot to 0.')
markerline, stemlines, baseline = plt.stem(x, y, '-')
if not color or color == 'None':
color = stemlines[0].get_color()
lolli_fmt = {'linewidth': lw, 'color': color}
pop_fmt = {'mec': color, 'markerfacecolor':'None', \
'markersize':10, 'markeredgewidth': lw*1.1}
plt.setp(markerline, **pop_fmt)
plt.setp(stemlines, **lolli_fmt)
plt.setp(baseline, 'color','k')
def test_lollipop():
"""Test function for lollipop()"""
from warnings import catch_warnings
t = np.arange(1, 20)*1e-3
f = 20.
a = np.sin(2*np.pi*f*t)
plt.figure()
with catch_warnings(record=True) as w:
lollipop(t, a, color=None, lw=1.5, ybot=0.1)
assert len(w) > 0
plt.grid(True)