Split data separated by comma into lists in python

Question

my dataframe df looks like this

Row_ID Codes
=============
1      A123,B456,C678
2      X359,C678,F23
3      J3,D24,J36,K994

I want to put all Codes in a list

something like this

['A123', 'B456', 'C678'],['X359', 'C678', 'F23'], ['J3', 'D24', 'J36', 'K994']

I did this

# an empty list
CodeList = [] 
for i in df['Codes']: 
  CodeList.append(list(i))

but what I get is this

['A','1','2','3','B'....

How can I do it the right way as mentioned above?

Remove list(i) from the line CodeList.append(list(i)). Just keep CodeList.append(i) — squareRoot17
– squareRoot17, Commented Nov 24, 2019 at 5:03
Variable names should follow the lower_case_with_underscores style, not camelCase or anything else. — AMC
– AMC, Commented Nov 24, 2019 at 8:59

Vlad Bezden · Accepted Answer · 2019-11-25 00:26:19Z

1

import pandas as pd


data = {"Codes": ["A123, B456, C678", "X359, C678, F23", "J3, D24, J36, K994"]}
df = pd.DataFrame(data)

result = [a.split(", ") for a in df["Codes"]]
print(result)

output

[['A123', 'B456', 'C678'], ['X359', 'C678', 'F23'], ['J3', 'D24', 'J36', 'K994']]

edited Nov 25, 2019 at 0:26

answered Nov 24, 2019 at 5:12

Vlad Bezden

90.7k27 gold badges261 silver badges190 bronze badges

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

4 Comments

AMC Over a year ago

The docs advise against using .values.

AMC Over a year ago

Also, is it just the way it was copied into the post or is that output list different from the one in OP’s post?

Vlad Bezden Over a year ago

@AlexanderCécile good point. I updated my result by removing values. So instead I'm just iterating over series values. Thank you.

AMC Over a year ago

Yeah I'm quite certain that this output is incorrect. OP's desired output is [['A123', 'B456', 'C678'],['X359', 'C678', 'F23'], ['J3', 'D24', 'J36', 'K994']].

Rik · Accepted Answer · 2019-11-25 01:13:30Z

1

Try spliting using the following:

CodeList.append(i.split(','))

edited Nov 25, 2019 at 1:13

answered Nov 24, 2019 at 4:19

Rik

1,9873 gold badges23 silver badges38 bronze badges

4 Comments

AMC Over a year ago

Why the extra [ ]?

Rik Over a year ago

@AlexanderCécile I think he wants array of arrays

AMC Over a year ago

I just tested it, the [ ] are unnecessary.

Rik Over a year ago

Ah ok, I'm thinking of += instead of append. thanks @AlexanderCécile

AMC · Accepted Answer · 2019-11-24 22:01:12Z

0

It seems like many of the other answers here might just be plain wrong. (Edit: Currently, they all are)

This code does work:

import pandas as pd

data = {'Codes': ['A123,B456,C678', 'X359,C678,F23', 'J3,D24,J36,K994']}
df = pd.DataFrame(data)

codes_list = df['Codes'].str.split(',').tolist()

codes_list looks like:

[['A123', 'B456', 'C678'], ['X359', 'C678', 'F23'], ['J3', 'D24', 'J36', 'K994']]

Note that this solution is idiomatic Pandas, whereas explicit loops should be avoided whenever possible.

edited Nov 24, 2019 at 22:01

answered Nov 24, 2019 at 9:04

AMC

2,6977 gold badges15 silver badges35 bronze badges

Comments

αԋɱҽԃ αмєяιcαη · Accepted Answer · 2019-11-24 04:36:23Z

0

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randn(3, 2), columns=list('AB'))
print(df.head())
print(df.values.tolist())

output:

[[-0.2645782053241853, 0.5022937587041725], [1.624868960959602, 0.5086915380333786], [1.3593608874498997, 0.7077939622903995]]

edited Nov 24, 2019 at 4:36

answered Nov 24, 2019 at 4:21

αԋɱҽԃ αмєяιcαη

11.6k3 gold badges23 silver badges58 bronze badges

5 Comments

AMC Over a year ago

The docs advise against using .values.

αԋɱҽԃ αмєяιcαη Over a year ago

@AlexanderCécile alright, it's usually good to have more than solution.

AMC Over a year ago

I said nothing to the contrary. We shouldn’t promote bad practices “because they’re different”, however, especially when there exists a good alternative. Also, I’m pretty sure that code doesn’t even produce the right result.

αԋɱҽԃ αмєяιcαη Over a year ago

@AlexanderCécile Okay

AMC Over a year ago

Yeah no i’m quite certain that this isn’t correct. Your code is literally just converting that DataFrame to a list.

Ashu007 · Accepted Answer · 2019-11-25 03:43:47Z

0

Just remove the list from line CodeList.append(list(i))

CodeList = [] 
for i in df['Codes']: 
   CodeList.append(i.split(','))

edited Nov 25, 2019 at 3:43

answered Nov 24, 2019 at 8:59

Ashu007

7951 gold badge9 silver badges14 bronze badges

2 Comments

AMC Over a year ago

Pretty sure this is incorrect. In any case, it should probably just be written df[‘Codes’].tolist().

AMC Over a year ago

I just checked, this is indeed incorrect. The output of CodeList will be ['A123, B456, C678', 'X359, C678, F23', 'J3, D24, J36, K994'].

Collectives™ on Stack Overflow

Split data separated by comma into lists in python

5 Answers 5

4 Comments

4 Comments

Comments

5 Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

5 Answers 5

4 Comments

4 Comments

Comments

5 Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related