Pandas convert dataframe to array of tuples with some special requirements

Question

I'm analysing some data with Apriori algorithm. This requires me to convert the dataframe into an array of tuples( groupby col"c1"), with each tuple corresponding to a "row" of the dataframe. If there's only one value in the tuple, I want an "," behind it cuz I need to keep it as a tuple.

In [1]: data
Out[1]:
   c1   c2
0  r1   aa
1  r1   bb
2  r1   cc
3  r2   dd
4  r2   ee
5  r3   ff

I expect the data like this:

[('aa','bb','cc'),('dd','ee'),('ff',)]

jezrael · Accepted Answer · 2019-04-14 16:04:20Z

3

Use GroupBy.apply with tuple and convert Series to list:

a = df.groupby('c1')['c2'].apply(tuple).tolist()
print (a)
[('aa', 'bb', 'cc'), ('dd', 'ee'), ('ff',)]

edited Apr 14, 2019 at 16:04

answered Apr 14, 2019 at 15:54

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Pandas convert dataframe to array of tuples with some special requirements

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related