迭代 numpy 数组并格式化为多维数组

问题描述

我正在尝试编写一个 numpy 函数，它将下面的 Appending_list 函数组织成一个多维数组，用于 Values 的每次迭代。所以在第一次迭代中，第一行是 Number_array + 0，第二行是 Number_array + 1，第三行是 Number_array + 2。

import numpy as np
from numpy import random
Values = np.arange(0,3,1)
Number_array =  np.arange(1,5,1)
Appending_list = Number_array + Values * len(Number_array)

预期输出

[[1,2,4],[2,4,5],[3,6]]

解决方法

values = np.arange(1,5)
rows = 3
# creates 3 rows of `[1,2,3,4]`
result = np.tile(values,rows).reshape((rows,len(values)))
shifts = np.arange(3)
# shifts the row values for each row by 0,1,and 2 respectively
result += shifts[:,None]

结果

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

,

您正在研究广播：

Appending_list = Number_array + Values[:,None]

输出：

array([[1,6]])