Extract a list of values from a column in a pandas dataframe

Question

I’m trying to extract a list of values from a column in a dataframe.

For example:

# dataframe with "num_fruit" column 
fruit_df = pd.DataFrame({"num_fruit": ['1 "Apple"', 
                                        '100 "Peach Juice3" 1234 "Not_fruit" 23 "Straw-berry" 2 "Orange"']})

# desired output: a list of values from the "num_fruit" column 
[['1 "Apple"'],
 ['100 "Peach Juice3"', '1234 "Not_fruit"', '23 "Straw-berry"', '2 "Orange"']]

Any suggestions? Thanks a lot.

What I’ve tried:

import re 

def split_fruit_val(val):
    return re.findall('(\d+ ".+")', val)

result_list = []
for val in fruit_df['num_fruit']:
    result = split_fruit_val(val)
    result_list.append(result)

print(result_list)

#output: some values were not split appropriately 
[['1 "Apple"'],
 ['100 "Peach Juice3" 1234 "Not_fruit" 23 "Straw-berry" 2 "Orange"']]

Shubham Sharma · Accepted Answer · 2023-01-20 06:25:13Z

3

Lets split with positive lookahead for a number

fruit_df['num_fruit'].str.split(r'\s(?=\d+)')

0                                          [1 "Apple"]
1    [100 "Peach Juice3", 1234 "Not_fruit", 23 "Str...
Name: num_fruit, dtype: object

answered Jan 20, 2023 at 6:25

Shubham Sharma

71.8k6 gold badges26 silver badges58 bronze badges

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

3 Comments

Grayrigel Over a year ago

A small addition:fruit_df['num_fruit'].str.split(r'\s(?=\d+)').values.tolist() could be added to match desired output.

Stella Over a year ago

Amazing. Which regex can I use if "Peach Juice3"is changed to ""Peach Juice 3 (250~500ml)"?

Shubham Sharma Over a year ago

@Stella I think something like this should work: fruit_df['num_fruit'].str.split(r'(?<=")\s(?=\d+)')

Collectives™ on Stack Overflow

Extract a list of values from a column in a pandas dataframe

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related