如何使用numpy.arange和另外两个数组作为开始和停止参数?

6kkfgxo0  于 2022-12-04  发布在  其他
关注(0)|答案(1)|浏览(120)

我想构建一个二维数组,其中第一维的长度与其他两个数组的长度相同,第二维是由numpy.arange创建的数组,并基于其他两个数组的每个元素,其中一个数组定义start参数,第二个数组定义stop参数-1。
让我给予一个例子:

arr_1 = array([0, 0, 0, 1, 2, 3, 4])
arr_2 = array([0, 1, 2, 3, 4, 5, 6])

我尝试创建一个如下的结果数组:

res_arr = array([[0], [0, 1], [0, 1, 2], [1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6]])

这能和numpy一起做吗?

ix0qys7i

ix0qys7i1#

您可以通过使用zip和列表解析来实现这一点

res_arr = [np.arange(start, stop + 1) for start, stop in zip(arr_1, arr_2)]

相关问题