Python生成大量随机人员信息

有时候我们会需要大量数据来演示或者测试某些功能，但是可能无法获得这些数据，或者担心泄露信息而不想使用真实数据。这时，我们可以自己生成一些随机数据。

import random

import string

def getEmail():

    # of course you can extend the following list

    suffix = ['.com', '.org', '.net', '.cn']

    characters = string.ascii_letters + string.digits + '_'

    username = ''.join((random.choice(characters) for i in range(random.randint(6,12))))

    domain = ''.join((random.choice(characters) for i in range(random.randint(3,6))))

    return username + '@' + domain + random.choice(suffix)

def getTelNo():

    return ''.join((str(random.randint(0,9)) for i in range(11)))

def getNameOrAddress(flag):

    '''flag=1 means Name,

       flag=0 meains Address'''

    #[0x4e00, 0x9fa5] is the range of Chinese character code 

    start, end = 0x4e00, 0x9fa5 

    result = ''

    #the length of most Chinese people's name is between 2 and 4

    if flag==1:

        rangestart, rangeend = 2, 5

    #here suppose the address contains at most 30 Chinese Characters

    elif flag==0:

        rangestart, rangeend = 10, 30

    else:

        print('flag must be 1 or 0')

        return

    for i in range(rangestart, rangeend):

        result += chr(random.randint(start, end))

    return result

def getSex():

    return random.choice(('男', '女'))

def getAge():

    return str(random.randint(18,100))

def main(filename):

    with open(filename, 'w', encoding='utf-8') as fp:

        fp.write('Name,Sex,Age,TelNO,Address,Email\n')

        #quickly generate information of 2000 persons

        for i in range(2000):

            name = getNameOrAddress(1)

            sex = getSex()

            age = getAge()

            tel = getTelNo()

            address = getNameOrAddress(0)

            email = getEmail()

            line = '{0},{1},{2},{3},{4},{5}\n'.format(name, sex, age, tel, address, email)

            fp.write(line)

def output(filename):

    with open(filename, 'r', encoding='utf-8') as fp:

        for line in fp:

            print(line.strip())

if __name__=='__main__':

    filename = 'information.txt'

    main(filename)

    output(filename)