Efficient way to create strings from a list

Question

I have some very inefficient code I'd like to make more general/efficient. I am trying to create strings from a set of lists.

Here is what I currently have:

#contains categories
numind = [('Length',), ('Fungus',)] 

#contains values that pertain to the categories
records = [('Length', 'Long'), ('Length', 'Med'), ('Fungus', 'Yes'), ('Fungus', 'No')] 

#contains every combination of values between the 2 categories. 
#for example, (Long, Yes) = Length=Long & Fungus = Yes.
combinations = [('Long', 'Yes'), ('Long', 'No'), ('Med', 'Yes'), ('Med', 'No')]

Now I want to create strings that have every combination in my combination list. This is the inefficient part. I'd like it so that I don't have to hardwire the length of the "numind" list. Any ideas?

values = combinations
valuestring = []

if len(numind) == 0:
   pass
elif len(numind) == 1:
   for a in xrange(len(values)):
      valuestring.append(numind[0][0]+values[a][0]) 

elif len(numind) == 2:
   for a in xrange(len(values)):
      valuestring.append(numind[0][0]+values[a][0]+'_'+numind[1][0]+values[a][1]) 

#and so forth until numind is 10+

output

['LengthLong_FungusYes', 'LengthLong_FungusNo', 'LengthMed_FungusYes', 'LengthMed_FungusNo']

DSM · Accepted Answer · 2013-08-27 19:25:25Z

I'd use itertools.product with collections.OrderedDict (the latter isn't strictly necessary, but means that you get the order right without having to think about it):

>>> from collections import OrderedDict
>>> from itertools import product
>>> 
>>> d = OrderedDict()
>>> for k, v in records:
...     d.setdefault(k, []).append(v)
...     
>>> d
OrderedDict([('Length', ['Long', 'Med']), ('Fungus', ['Yes', 'No'])])
>>> ['_'.join(k+v for k,v in zip(d, v)) for v in product(*d.values())]
['LengthLong_FungusYes', 'LengthLong_FungusNo', 'LengthMed_FungusYes', 'LengthMed_FungusNo']

itertools.product naturally produces the "every combination" part (which is really called the Cartesian product, not a combination):

>>> product(["Long", "Med"], ["Yes", "No"])
<itertools.product object at 0x96b0dec>
>>> list(product(["Long", "Med"], ["Yes", "No"]))
[('Long', 'Yes'), ('Long', 'No'), ('Med', 'Yes'), ('Med', 'No')]

The advantage here is that it doesn't matter how many categories there are or how many values there are associated with any category: as long as they're specified in records, it should work.

wflynny · Accepted Answer · 2013-08-27 19:22:15Z

0

Try using itertools

s = 'Length%s_Fungus%s'
l1 = ['Long', 'Med']
l2 = ['Yes', 'No']
[s%x for x in itertools.product(l1, l2)]

This yields

['LenLong_FungusYes',
 'LenMed_FungusNo',
 'LenMed_FungusYes',
 'LenLong_FungusNo']

answered Aug 27, 2013 at 19:22

wflynny

18.6k6 gold badges50 silver badges69 bronze badges

Comments

Andrew Clark · Accepted Answer · 2013-08-27 19:25:51Z

0

>>> numind = [('Length',), ('Fungus',)]
>>> combinations = [('Long', 'Yes'), ('Long', 'No'), ('Med', 'Yes'), ('Med', 'No')] 
>>> ['_'.join(numind[i][0] + v for i, v in enumerate(c)) for c in combinations]
['LengthLong_FungusYes', 'LengthLong_FungusNo', 'LengthMed_FungusYes', 'LengthMed_FungusNo']

answered Aug 27, 2013 at 19:25

Andrew Clark

210k36 gold badges286 silver badges310 bronze badges

Comments

CT Zhu · Accepted Answer · 2013-08-27 19:30:55Z

0

How general you want it to be?

>>> for item in combinations:
    '_'.join(map(''.join, zip(list(itertools.chain(*numind)), item)))


'LengthLong_FungusYes'
'LengthLong_FungusNo'
'LengthMed_FungusYes'
'LengthMed_FungusNo'

answered Aug 27, 2013 at 19:30

CT Zhu

54.6k18 gold badges125 silver badges136 bronze badges

Collectives™ on Stack Overflow

Efficient way to create strings from a list

4 Answers 4

Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related