Python list comprehension - generating multiple data

Question

import random

M = 4
N = 3


def generisanjevol1(nekalista, m):
    return random.choices(nekalista, k=m)


def generisanjevol2(nekalista, m,n):
    #obj = [[random.choice(nekalista,k=m)] for i in range(N)]]
    obj = [[random.choice(nekalista)] for i in range(n)]
    return obj

    #def poredjenje()


listaslova = ['A', 'B', 'C', 'D', 'E']

lista = generisanjevol1(listaslova, M)
lista2 = generisanjevol2(listaslova, M, N)
print(lista)
print(lista2)

So above is my try (generisanjevol2(nekalista, m,n)...

What I am trying to do is next: I want to generate N of arrays and fill them with strings which are generated by random.choice function and they still must be strings from listaslova)
Perhaps let's say N=3 (N represents numbers of arrays) and M=4 (M represents length of array) I should get something like this (doesn't have to be same data in arrays, because of course they are randomly generated):

[A,C,D,E]
([A,C,E,D] [E,C,B,A] [E,D,D,A])

But the results which I get are following:

[A,D,E,C]
[[B],[D],[E]]

P.S If I try the one which is commented I get an error

Barmar · Accepted Answer · 2020-10-10 19:13:00Z

1

The error in your commented line is because you have an extra ]. And random.choice should be random.choices.

But you also shouldn't put another list around the call to random.choice(). It already returns a list.

def generisanjevol2(nekalista, m,n):
    obj = [random.choices(nekalista,k=m) for i in range(n)]
    return obj

edited Oct 10, 2020 at 19:13

answered Oct 10, 2020 at 19:10

Barmar

789k57 gold badges555 silver badges669 bronze badges

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

14 Comments

str1ng Over a year ago

In commented line I get error with random.choice(nekalista, k=m)... It says following: TypeError: choice() got an unexpected keyword argument 'k'

Barmar Over a year ago

It should be random.choices, just like in generisanjevol1

Barmar Over a year ago

By calling random.choices() in generisanjevol2, like I showed in the answer.

Barmar Over a year ago

Did you see that I edited the answer after you posted your comment?

Barmar Over a year ago

Write a function that compares the lists and returns a score of how similar they are. Then you can use sorted(zip(lista2, lista2), key=lambda pair: yourFunction(*pair)).

|

Ilan Pe'er · Accepted Answer · 2020-10-10 19:19:19Z

1

Like @Barmar said, you indeed have an extra []. Your function should look:

def generisanjevol2(nekalista, m,n):
...:     obj = [random.choices(nekalista, k=m) for i in range(n)]
...:     return obj

answered Oct 10, 2020 at 19:19

Ilan Pe'er

314 bronze badges

Collectives™ on Stack Overflow

Python list comprehension - generating multiple data

2 Answers 2

14 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

14 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related