How do I read a random line from one file?

Question:

Is there a built-in method to do it? If not how can I do this without costing too much overhead?

Asked By: Shane

||

Answers:

Seek to a random position, read a line and discard it, then read another line. The distribution of lines won’t be normal, but that doesn’t always matter.

Not built-in, but algorithm R(3.4.2) (Waterman’s "Reservoir Algorithm") from Knuth’s "The Art of Computer Programming" is good (in a very simplified version):

import random

def random_line(afile):
    line = next(afile)
    for num, aline in enumerate(afile, 2):
        if random.randrange(num):
            continue
        line = aline
    return line

The num, ... in enumerate(..., 2) iterator produces the sequence 2, 3, 4… The randrange will therefore be 0 with a probability of 1.0/num — and that’s the probability with which we must replace the currently selected line (the special-case of sample size 1 of the referenced algorithm — see Knuth’s book for proof of correctness == and of course we’re also in the case of a small-enough "reservoir" to fit in memory ;-))… and exactly the probability with which we do so.

Answered By: Alex Martelli

It depends what do you mean by “too much” overhead. If storing whole file in memory is possible, then something like

import random

random_lines = random.choice(open("file").readlines())

would do the trick.

Answered By: cji
import random
lines = open('file.txt').read().splitlines()
myline =random.choice(lines)
print(myline)

For very long file:
seek to random place in file based on it’s length and find two newline characters after position (or newline and end of file). Do again 100 characters before or from beginning of file if original seek position was <100 if we ended up inside the last line.

However this is over complicated, as file is iterator.So make it list and take random.choice (if you need many, use random.sample):

import random
print(random.choice(list(open('file.txt'))))
Answered By: Tony Veijalainen

Although I am four years late, I think I have the fastest solution. Recently I wrote a python package called linereader, which allows you to manipulate the pointers of file handles.

Here is the simple solution to getting a random line with this package:

from random import randint
from linereader import dopen

length = #lines in file
filename = #directory of file

file = dopen(filename)
random_line = file.getline(randint(1, length))

The first time this is done is the worst, as linereader has to compile the output file in a special format. After this is done, linereader can then access any line from the file quickly, whatever size the file is.

If your file is very small (small enough to fit into an MB), then you can replace dopen with copen, and it makes a cached entry of the file within memory. Not only is this faster, but you get the number of lines within the file as it is loaded into memory; it is done for you. All you need to do is to generate the random line number. Here is some example code for this.

from random import randint
from linereader import copen

file = copen(filename)
lines = file.count('n')
random_line = file.getline(randint(1, lines))

I just got really happy because I saw someone who could benefit from my package! Sorry for the dead answer, but the package could definitely be applied to many other problems.

Answered By: Nick Pandolfi

If you don’t want to read over the entire file, you can seek into the middle of the file, then seek backwards for the newline, and call readline.

Here is a Python3 script which does just this,

One disadvantage with this method is short lines have lower likelyhood of showing up.

def read_random_line(f, chunk_size=16):
    import os
    import random
    with open(f, 'rb') as f_handle:
        f_handle.seek(0, os.SEEK_END)
        size = f_handle.tell()
        i = random.randint(0, size)
        while True:
            i -= chunk_size
            if i < 0:
                chunk_size += i
                i = 0
            f_handle.seek(i, os.SEEK_SET)
            chunk = f_handle.read(chunk_size)
            i_newline = chunk.rfind(b'n')
            if i_newline != -1:
                i += i_newline + 1
                break
            if i == 0:
                break
        f_handle.seek(i, os.SEEK_SET)
        return f_handle.readline()
Answered By: ideasman42

You can add the lines into a set() which will change their order randomly.

filename=open("lines.txt",'r')
f=set(filename.readlines())
filename.close()

To find the 1st line:

print(next(iter(f)))

To find the 3rd line:

print(list(f)[2])

To list all the lines in the set:

for line in f:
    print(line)
Answered By: GoTrained

This may be bulky, but it works I guess? (at least for txt files)

import random
choicefile=open("yourfile.txt","r")
linelist=[]
for line in choicefile:
    linelist.append(line)
choice=random.choice(linelist)
print(choice)

It reads each line of a file, and appends it to a list. It then chooses a random line from the list.
If you want to remove the line once it’s chosen, just do

linelist.remove(choice)

Hope this may help, but at least no extra modules and imports (apart from random) and relatively lightweight.

Answered By: Philip Hughes
import random

with open("file.txt", "r") as f:
    lines = f.readlines()
    print (random.choice(lines))
Answered By: HCLivess

A slightly improved version of the Alex Martelli’s answer, which handles empty files (by returning a default value):

from random import randrange

def random_line(afile, default=None):
    line = default
    for i, aline in enumerate(afile, start=1):
        if randrange(i) == 0:  # random int [0..i)
            line = aline
    return line

This approach can be used to get a random item from any iterator using O(n) time and O(1) space.

Answered By: Eugene Yarmash

If you don’t want to load the whole file into RAM with f.read() or f.readlines(), you can get random line this way:

import os
import random


def get_random_line(filepath: str) -> str:
    file_size = os.path.getsize(filepath)
    with open(filepath, 'rb') as f:
        while True:
            pos = random.randint(0, file_size)
            if not pos:  # the first line is chosen
                return f.readline().decode()  # return str
            f.seek(pos)  # seek to random position
            f.readline()  # skip possibly incomplete line
            line = f.readline()  # read next (full) line
            if line:
                return line.decode()  
            # else: line is empty -> EOF -> try another position in next iteration

P.S.: yes, that was proposed by Ignacio Vazquez-Abrams in his answer above, but a) there’s no code in his answer and b) I’ve come up with this implementation myself; it can return first or last line. Hope it may be useful for someone.

However, if you care about distribution, this code is not an option for you.

Answered By: Ivan Vinogradov

Here is another way, a little like Philip Hughes’ explanation, but with an addition of .strip in case you are collecting a random line that needs extra space cleaning or for jumping rows.

Code:

import random

def random_line():
    file = open("file.txt", "r")
    text = []

    for line in file:
        line = line.strip() 
        text.append(line)

    file.close()

    x = random.randrange(0,len(text))
    return x

It does the following: imports the random module, reads a file, creates a list with the extracted and "cleaned" lines from that file (.txt in this case), closes the file, and selects a random item (that was a line in the .txt) from the created list.

Answered By: Rhappy
Categories: questions Tags:
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.