Make new variable by if condition in python

Question

I have a list in python :

['ABCD_BAND1.TXT', 'ABCD_BAND2.TXT', 'ABCD_BAND3.TXT', 'ABCD_BAND4.TXT', 'ABCD_BAND5.TXT', 'ABCD_BAND6.TXT', 'ABCD_BAND7.TXT']

I want to pick ABCD_BAND4.TXT & ABCD_BAND5.TXT and define them as two new variables and do my analysis based on these two variables. I have written the following code but it does not works

Assign Variables

dataList = ['ABCD_BAND1.TXT', 'ABCD_BAND2.TXT', 'ABCD_BAND3.TXT', 'ABCD_BAND4.TXT', 'ABCD_BAND5.TXT', 'ABCD_BAND6.TXT', 'ABCD_BAND7.TXT']
for item in dataList:
    if item.endswith("BAND4.TXT"):
        Item = "band4"
for item in dataList:
    if item.endswith("BAND5.TXT"):
        Item = "band5"

How can I fix it?

Cheers

What seems to be the problem with this? Item is first set to "band4" in the first loop and then Item is set to "band5" in the second loop. Do you just need to use two different names, say item1 and item2? — AChampion
– AChampion, Commented Sep 9, 2015 at 3:59

AChampion · Accepted Answer · 2015-09-09 04:06:06Z

3

You need to declare two separate variables for storing the returned output.

>>> s = ['ABCD_BAND1.TXT', 'ABCD_BAND2.TXT', 'ABCD_BAND3.TXT', 'ABCD_BAND4.TXT', 'ABCD_BAND5.TXT', 'ABCD_BAND6.TXT', 'ABCD_BAND7.TXT']

>>> item4 , item5 = "",""
>>> for i in s:
        if i.endswith("BAND4.TXT"):
            item4 = i
        elif i.endswith("BAND5.TXT"):
            item5 = i

>>> item4
'ABCD_BAND4.TXT'
>>> item5
'ABCD_BAND5.TXT'
>>>

edited Sep 9, 2015 at 4:06

AChampion

30.5k4 gold badges63 silver badges79 bronze badges

answered Sep 9, 2015 at 3:49

Avinash Raj

175k32 gold badges247 silver badges289 bronze badges

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

Comments

The6thSense · Accepted Answer · 2015-09-09 04:58:55Z

2

You can use result_list depend on your purpose.

raw_list = ['ABCD_BAND1.TXT', 'ABCD_BAND2.TXT', 'ABCD_BAND3.TXT','ABCD_BAND4.TXT', 'ABCD_BAND5.TXT', 'ABCD_BAND6.TXT', 'ABCD_BAND7.TXT']

chose_list =  ['ABCD_BAND4.TXT', 'ABCD_BAND5.TXT']

result_list = [item for item in raw_list if item in chose_list]

edited Sep 9, 2015 at 4:58

The6thSense

8,3559 gold badges38 silver badges67 bronze badges

answered Sep 9, 2015 at 4:18

Nguyen Quang Trung

1055 bronze badges

Comments

Trigremm · Accepted Answer · 2015-09-09 04:06:18Z

0

this solution works for me

>>> DataList = ['ABCD_BAND1.TXT', 'ABCD_BAND2.TXT', 'ABCD_BAND3.TXT', 'ABCD_BAND4.TXT', 'ABCD_BAND5.TXT', 'ABCD_BAND6.TXT', 'ABCD_BAND7.TXT']

>>> band4 = [i for i in DataList if "BAND4.TXT" in i][0] # 0 index for first occurance in list
>>> band4 
'ABCD_BAND4.TXT'
>>> band5 = [i for i in DataList if "BAND5.TXT" in i][0] # 0 index for first occurance in list
>>> band5
'ABCD_BAND5.TXT'

answered Sep 9, 2015 at 4:06

Trigremm

717 bronze badges

2 Comments

Green Cell Over a year ago

Looping through the array twice is not exactly efficient.

Trigremm Over a year ago

I have just improved authors code to work, and there was no efficiency requirement.

Collectives™ on Stack Overflow

Make new variable by if condition in python

Assign Variables

3 Answers 3

Comments

Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

Assign Variables

3 Answers 3

Comments

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related