Explode array [(str), (int)] in column dataframe pandas

Question

I have a dataframe:

    df = pd.DataFrame({
       'day': ['11', '12'],
       'City': ['[(Mumbai, 1),(Bangalore, 2)]', '[(Pune, 3),(Mumbai, 4),(Delh, 5)]']
    })

   day                               City
0  11       [(Mumbai, 1),(Bangalore, 2)]
1  12  [(Pune, 3),(Mumbai, 4),(Delh, 5)]

I want to make an explode. But when I do that, nothing changes.

df2 = df.explode('City')

What I want to get at the output

  day            City
0  11     (Mumbai, 1)
1  11  (Bangalore, 2)
2  12       (Pune, 3)
3  12     (Mumbai, 4)
4  12       (Delh, 5)

You have strings, you need lists to explode. What is the wanted output? a string or tuple? — mozway
– mozway, Commented Mar 9, 2022 at 16:11

mozway · Accepted Answer · 2022-03-09 16:18:12Z

2

You can explode strings. You need to find a way to convert to lists.

Assuming you have city names with only letters (or spaces) you could use a regex to add the quotes and convert to list with ast.literal_eval:

from ast import literal_eval

df['City'] = (df['City']
              .str.replace(r'([a-zA-Z ]+),', r'"\1",', regex=True)
              .apply(literal_eval)
              )

df2 = df.explode('City', ignore_index=True)

output:

  day            City
0  11     (Mumbai, 1)
1  11  (Bangalore, 2)
2  12       (Pune, 3)
3  12     (Mumbai, 4)
4  12       (Delh, 5)

answered Mar 9, 2022 at 16:18

mozway

267k13 gold badges56 silver badges106 bronze badges

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

Comments

Moun · Accepted Answer · 2022-03-09 16:33:00Z

0

df = pd.DataFrame({
    'day': ['11', '12'],
    'City': ['[(Mumbai, 1),(Bangalore, 2)]', '[(Pune, 3),(Mumbai, 4),(Delh, 5)]']
})


df['City'] = [re.sub("\),\(",")-(", x) for x in df['City']]
df['City'] = [re.sub("\[|\]|\(|\)","", x) for x in df['City']]
df['City'] = [x.split("-") for x in df['City']]
df['City']
df2 = df.explode('City').reset_index(drop=True)

you have to process the string and convert it to list before explode

  day          City
0  11     Mumbai, 1
1  11  Bangalore, 2
2  12       Pune, 3
3  12     Mumbai, 4
4  12       Delh, 5

answered Mar 9, 2022 at 16:33

Moun

3253 silver badges23 bronze badges

Collectives™ on Stack Overflow

Explode array [(str), (int)] in column dataframe pandas

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related