One-liner to create multiple strings from single loop in python

Question

Using list comprehension I can create something simple like this

['X{}'.format(x) for x in range(2)]

which produces ['X0', 'X1']. But I want to create separate and multiple output strings, preferably from the same list comprehension statement.

I want my output to be ['X0','Y0','Z0','X1','Y1' 'Z1'] naturally something like ['X{},Y{},Z{}'.format(x,x,x) for x in range(2)], doesn't quite cut the mustard.

Any ideas of how to do this in one line?

EDIT: It is not paramount that it is on one line, but it would be nice.

Moses Koledoye · Accepted Answer · 2017-01-14 00:16:27Z

5

Here's one way to do it using a nested comprehension:

r = ['{}{}'.format(ch, x) for x in range(2) for ch in 'XYZ']
print(r)
# ['X0', 'Y0', 'Z0', 'X1', 'Y1', 'Z1']

answered Jan 14, 2017 at 0:16

Moses Koledoye

78.8k8 gold badges139 silver badges141 bronze badges

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

Comments

blue note · Accepted Answer · 2017-01-14 00:17:28Z

3

Double list comprehension

reduce([ letter+str(number) for letter in 'ABC' for number in range(x)], [])

answered Jan 14, 2017 at 0:17

blue note

29.5k10 gold badges83 silver badges110 bronze badges

Comments

DYZ · Accepted Answer · 2017-01-14 00:17:17Z

1

And one more way to go (still a one-liner):

sum([['X{}'.format(x),'Y{}'.format(x),'Z{}'.format(x)] for x in range(2)],[])
# ['X0', 'Y0', 'Z0', 'X1', 'Y1', 'Z1']

answered Jan 14, 2017 at 0:17

DYZ

57.3k10 gold badges73 silver badges101 bronze badges

Comments

inspectorG4dget · Accepted Answer · 2017-01-14 00:19:11Z

1

With a little help from itertools

In [59]: ["{}{}".format(char,i) for i,char in itertools.product(range(2), "XYZ")]
Out[59]: ['X0', 'Y0', 'Z0', 'X1', 'Y1', 'Z1']

answered Jan 14, 2017 at 0:19

inspectorG4dget

115k30 gold badges159 silver badges253 bronze badges

Comments

Imperial_Squid · Accepted Answer · 2017-01-14 00:28:39Z

0

Try this:

List = ['{}{}'.format(Letter, Num) for Num in range(2) for Letter in 'XYZ']
print(List)

answered Jan 14, 2017 at 0:28

Imperial_Squid

178 bronze badges

Collectives™ on Stack Overflow

One-liner to create multiple strings from single loop in python

5 Answers 5

Comments

Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

5 Answers 5

Comments

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related