首页 > 解决方案 > 创建 N 个长度为 M 的数组,其中序列在每个数组上增加 1

问题描述

早上好,

给定两个数字(a,b),我应该创建长度为“b”的“a”个不同列表,其中每个列表的序列加一。

例如 :

1,2 =>[[0,1]]

2,2 = >[[0,1], [1,2]]

我正在尝试编写以下函数:

def increase_arrays(arrays, length):
    result = [[i for i in range(length)] for i in range(arrays)]
    return result

increase_arrays(2,3)
=> [[0, 1, 2], [0, 1, 2]]

我看不到如何修改我的代码,所以第二个数组是[1,2,3]. 任何人都可以帮助解决这个问题吗?

标签: pythonarrays

解决方案


你有著名的重复索引问题,你有 2 个索引i

以下是您需要的代码:

def increase_arrays(arrays, length):
    result = [[i + j for i in range(length)] for j in range(arrays)]
    return result

increase_arrays(2, 3)返回[[0, 1, 2], [1, 2, 3]]


推荐阅读