How to concatenate output data coming out from for loop/function

Question

I have a data set as a list. From that list I need to find a keyword, and if found, I should be able to extract the complete information of that element.

att =['Email/[email protected]', 'CountryCode/US','CountryCode/UK', 'ID/12345']

from the above list, if I search only CountryCode, then the output should come as:

['CountryCode/US','CountryCode/UK']

Below is my code which I am trying, but it is returning only a single value.

Can some one help me with this code, to return all of the values as :

['CountryCode/US','CountryCode/UK']

def att_func(field,data):
    for i, j in enumerate(data):
        # print(i,j)
        if field in j:
            return [data[i]]


att =['Email/[email protected]', 'CountryCode/US','CountryCode/UK', 'ID/12345']

field ='CountryCode'

Cntry = att_func('CountryCode',att)
print(Cntry)

There is no error, but I am not getting the desired output, This code is giving output as ['CountryCode/US'] only.... — ankit
– ankit, Commented Aug 13, 2019 at 12:08
yes.. I got it. This is because you are returning as soon as single data is getting. I have added answer by modifying your code. — Harsha Biyani
– Harsha Biyani, Commented Aug 13, 2019 at 12:12

Sayse · Accepted Answer · 2019-08-13 12:08:04Z

2

You're returning early in your for loop so you'll never get past the first position where the field exists, you can either use yield instead of return and make your function return a generator or just use a list comprehension

def att_func(field, data):
    return [i for i in data if field in i]

answered Aug 13, 2019 at 12:08

Sayse

43.4k14 gold badges85 silver badges150 bronze badges

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

1 Comment

Chris Doyle Over a year ago

Excellent answer as list comprehension makes this so neat and clean.

Harsha Biyani · Accepted Answer · 2019-08-13 12:14:20Z

0

Reason : This is because you are returning as soon as single data is getting.

Solution : Collect all data in list and the return collected data.

Try:

def att_func(field,data):
    final_data = []
    for i, j in enumerate(data):
        # print(i,j)
        if field in j:
            final_data.append(data[i])
    return final_data


att =['Email/[email protected]', 'CountryCode/US','CountryCode/UK', 'ID/12345']

field ='CountryCode'

Cntry = att_func('CountryCode',att)
print(Cntry)

edited Aug 13, 2019 at 12:14

answered Aug 13, 2019 at 12:08

Harsha Biyani

7,28810 gold badges42 silver badges64 bronze badges

Comments

Valentin B. · Accepted Answer · 2019-08-13 12:14:38Z

0

att = ['Email/[email protected]', 'CountryCode/US','CountryCode/UK', 'ID/12345']
out = [item for item in att if "CountryCode" in item]

Done !

Your att_func should look like this:

def att_func(needle, haystack):
    return [item for item in haystack if needle in item]

Collectives™ on Stack Overflow

How to concatenate output data coming out from for loop/function

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related