我想构建一个二维数组,其中第一维的长度与其他两个数组的长度相同,第二维是由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一起做吗?
1条答案
按热度按时间ix0qys7i1#
您可以通过使用zip和列表解析来实现这一点