帮我用python做一个 自定义长度的随机函数，可以选择随机 长度和数字，或者数字字母结合

您好，以下是一个实现您需求的 Python 代码示例：

import random
import string

def generate_random_string(length, digits_only=True):
    """
    根据指定长度生成随机字符串

    参数：
    length: int，生成字符串的长度
    digits_only: bool，是否只包含数字，True表示只包含数字，False表示包含数字和字母

    返回值：
    随机生成的字符串
    """
    if digits_only:
        # 只包含数字的情况
        return ''.join(random.choices(string.digits, k=length))
    else:
        # 包含数字和字母的情况
        alphanumeric = string.ascii_letters + string.digits
        return ''.join(random.choices(alphanumeric, k=length))

# 示例调用
print(generate_random_string(10, True))  # 生成长度为10的随机数字字符串
print(generate_random_string(8, False))  # 生成长度为8的随机数字和字母组合字符串

代码中，`generate_random_string` 函数接受两个参数：`length` 表示需要生成的字符串长度，`digits_only` 表示生成的字符串是否只包含数字。函数内部通过 `random.choices` 函数和 `string.digits` 或 `string.ascii_letters` 生成随机字符串。最后在示例中分别调用函数，生成长度为 10 的随机数字字符串和长度为 8 的随机数字和字母组合字符串。

希望能够帮到您！