Randomly generate 2D list in Python

Question

I am trying to generate 5x5 list that has exactly 10 ones placed in a random locations in the 2D list.

I want to make the rest of the entries is zeroes. How can I make it?

import random

def randomNumbers():

    mylist=[random.randint(0, 1) for _ in range(5)]
    return mylist

Answer 1

You can do this

from random import shuffle
def randomNumbers():
    l=[1 for _ in range(10)]+[0 for _ in range(15)]
    shuffle(l)
    lst=[]
    for i in range(0,25,5):
        lst.append(l[i:i+5])
    return lst