创建N个长度为M的数组，其中每个数组的序列加1

Question

Good morning,

Given two numbers (a,b), I should create 'a' different list of length 'b' where the sequence of each list is increased by one.

For example :

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

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

I am trying to write the following function:

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]]

I can't see how to modify my code so the second array is [1,2,3] . Could anyone help to resolve the issue?

Answer 1

You have famous duplicated index problem, you have 2 indexes i .

Below is the code that you need:

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

increase_arrays(2, 3) returns [[0, 1, 2], [1, 2, 3]]

Answer 2

I finally managed to solve it with:

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

Answer 3

这是一个很有趣的解决方案，只是为了好玩：

np.arange(arrays)[:,np.newaxis] + np.arange(length)[np.newaxis,:]