简体   繁体   中英

Function to randomly read a line from a text file

I have to create a function that reads a random line from a text file in python.

I have the following code but am not able to get it to work

import random

def randomLine(filename):
    #Retrieve a  random line from a file, reading through the file irrespective of the length
    fh = open(filename.txt, "r")
    lineNum = 0
    it = ''

    while 1:
        aLine = fh.readline()
        lineNum = lineNum + 1
        if aLine != "":

            # How likely is it that this is the last line of the file ? 
            if random.uniform(0,lineNum)<1:
                it = aLine
        else:
            break

    fh.close()

    return it
print(randomLine(testfile.txt))

I got so far but,need help to go further, Please help

once the program is running i'm getting an error saying

print(randomLine(testfile.txt))
NameError: name 'testfile' is not defined

Here's a version that's been tested to work, and avoids empty lines.

Variable names are verbose for clarity.

import random
import sys

def random_line(file_handle):
    lines = file_handle.readlines()
    num_lines = len(lines)

    random_line = None
    while not random_line:
        random_line_num = random.randint(0, num_lines - 1)
        random_line = lines[random_line_num]
        random_line = random_line.strip()

    return random_line

file_handle = None

if len(sys.argv) < 2:
    sys.stderr.write("Reading stdin\n")
    file_handle = sys.stdin
else:
    file_handle = open(sys.argv[1])

print(random_line(file_handle))

file_handle.close()

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM