From c420c482e5fcccd28d897987d74cff514fada78f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bertrand=20N=C3=A9ron?= <bneron@pasteur.fr>
Date: Fri, 28 Apr 2017 10:32:48 +0200
Subject: [PATCH] change split_data function to use groupby
---
craw/heatmap.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/craw/heatmap.py b/craw/heatmap.py
index d820061..13a7719 100644
--- a/craw/heatmap.py
+++ b/craw/heatmap.py
@@ -52,11 +52,10 @@ def split_data(data):
:param data: the coverage data to split
:type data: a 2 dimension :class:`pandas.DataFrame` object
: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']
- antisense = data.loc[data['sense'] == 'AS']
- return sense, antisense
+ grp = data.groupby(by=['sense'])
+ return grp.get_group('S').copy(), grp.get_group('AS').copy()
def sort(data, criteria, **kwargs):
--
GitLab