Skip to content
Snippets Groups Projects
Commit c420c482 authored by Bertrand NÉRON's avatar Bertrand NÉRON
Browse files

change split_data function to use groupby

parent e714770c
No related branches found
No related tags found
No related merge requests found
Pipeline #
Hide whitespace changes
Inline Side-by-side
Showing
with 3 additions and 4 deletions
craw/heatmap.py
+ 3
− 4
View file @ c420c482
...@@ -52,11 +52,10 @@ def split_data(data): ...@@ -52,11 +52,10 @@ def split_data(data):
:param data: the coverage data to split :param data: the coverage data to split
:type data: a 2 dimension :class:`pandas.DataFrame` object :type data: a 2 dimension :class:`pandas.DataFrame` object
:return: two matrix :return: two matrix
:rtype: tuple of two :class:`pandas.DataFrame` object :rtype: tuple of two :class:`pandas.DataFrame` object (sense pandas.DataFrame, antisense pandas.DataFrame)
""" """
sense = data.loc[data['sense'] == 'S'] grp = data.groupby(by=['sense'])
antisense = data.loc[data['sense'] == 'AS'] return grp.get_group('S').copy(), grp.get_group('AS').copy()
return sense, antisense
def sort(data, criteria, **kwargs): def sort(data, criteria, **kwargs):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment