Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
Vasudevan, Rama K
pycroscopy
Commits
fc0ea3ac
Commit
fc0ea3ac
authored
Aug 10, 2017
by
Somnath, Suhas
Committed by
CompPhysChris
Aug 14, 2017
Browse files
Simplified the plot function for cluster visulization
parent
7cf243cd
Changes
1
Hide whitespace changes
Inline
Side-by-side
pycroscopy/viz/plot_utils.py
View file @
fc0ea3ac
...
...
@@ -19,7 +19,7 @@ from scipy.signal import blackman
from
matplotlib.colors
import
LinearSegmentedColormap
from
mpl_toolkits.axes_grid1
import
ImageGrid
from
..io.hdf_utils
import
reshape_to_Ndims
,
get_formatted_labels
from
..io.hdf_utils
import
reshape_to_Ndims
,
get_formatted_labels
,
get_data_descriptor
if
sys
.
version_info
.
major
==
3
:
unicode
=
str
...
...
@@ -702,7 +702,7 @@ def plot_map_stack(map_stack, num_comps=9, stdevs=2, color_bar_mode=None, evenly
return
fig202
,
axes202
def
plot_cluster_h5_group
(
h5_group
,
y_spec_label
,
centroids_together
=
True
):
def
plot_cluster_h5_group
(
h5_group
,
centroids_together
=
True
):
"""
Plots the cluster labels and mean response for each cluster
...
...
@@ -710,8 +710,6 @@ def plot_cluster_h5_group(h5_group, y_spec_label, centroids_together=True):
----------
h5_group : h5py.Datagroup object
H5 group containing the labels and mean response
y_spec_label : str
Label to use for Y axis on cluster centroid plot
centroids_together : Boolean, optional - default = True
Whether or nor to plot all centroids together on the same plot
...
...
@@ -722,7 +720,7 @@ def plot_cluster_h5_group(h5_group, y_spec_label, centroids_together=True):
axes : 1D array_like of axes objects
Axes of the individual plots within `fig`
"""
# TODO: The quantity and units for the main dataset itself are missing in most cases!
h5_labels
=
h5_group
[
'Labels'
]
try
:
h5_mean_resp
=
h5_group
[
'Mean_Response'
]
...
...
@@ -756,6 +754,8 @@ def plot_cluster_h5_group(h5_group, y_spec_label, centroids_together=True):
# Figure out the correct axes labels for label map:
pos_labels
=
get_formatted_labels
(
h5_pos_vals
)
y_spec_label
=
get_data_descriptor
(
h5_mean_resp
)
# TODO: cleaner x and y axes labels instead of 0.0000125 etc.
if
centroids_together
:
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment