📅  最后修改于: 2023-12-03 15:42:24.443000             🧑  作者: Mango
在进行数据分析、机器学习或其他计算机科学领域的学习和实践时,一种很常见的需要是随机创建一些大小确定的列表。本文将介绍一个 Python 程序,用于随机创建 N 个 K 大小的列表。
import random
def create_lists(n, k):
"""
随机创建 N 个 K 大小的列表
:param n: 创建的列表数量
:param k: 每个列表的大小
:return: 随机生成的 N 个 K 大小的列表
"""
lists = []
for i in range(n):
lst = [random.randint(0, 100) for _ in range(k)]
lists.append(lst)
return lists
该程序定义了一个名为 create_lists
的函数,该函数接受两个参数 n
和 k
,分别代表需要创建的列表数量和每个列表的大小。
程序使用 Python 自带的 random
模块生成随机数,并使用列表推导式创建大小为 K 的列表。最后将所有创建的列表添加到一个列表中返回。
下面是一个使用该程序的示例:
# 创建 3 个大小为 5 的列表
lists = create_lists(3, 5)
print(lists)
输出:
[[41, 79, 5, 12, 20], [54, 92, 34, 78, 4], [2, 50, 71, 27, 98]]
可以看到,程序成功随机创建了 3 个大小为 5 的列表。