将两个列表转换为矩阵

Question

我会尽量清楚，我将首先解释为什么我要将两个数组转换为矩阵。

为了绘制投资组合与市场指数的表现，我需要一种类似这种格式的数据结构：

[[portfolio_value1, index_value1]
 [portfolio_value2, index_value2]]

但我将数据作为两个独立的1-D阵列：

portfolio = [portfolio_value1, portfolio_value2, ...]
index = [index_value1, index_value2, ...]

那么如何将第二个场景转换为第一个场景呢？我已经尝试np.insert将第二个数组添加到我在python shell中的测试矩阵中，我的问题是将第一个数组转换为单个列矩阵。

如果没有命令性循环，如何实现这一目标的任何帮助都会很棒。

Answer 1

您想要的标准numpy函数是np.column_stack：

>>> np.column_stack(([1, 2, 3], [4, 5, 6]))
array([[1, 4],
       [2, 5],
       [3, 6]])

使用portfolio和index数组，执行

np.column_stack((portfolio, index))

会产生类似的东西：

[[portfolio_value1, index_value1],
 [portfolio_value2, index_value2],
 [portfolio_value3, index_value3],
 ...]

Answer 2

您可以使用np.c _

np.c_[[1,2,3], [4,5,6]]

它将给您：

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

Answer 3

假设投资组合和指数的长度相同：

matrix = []
for i in range(len(portfolio)):
    matrix.append([portfolio[i], index[i]])

或使用列表理解的单行：

matrix2 = [[portfolio[i], index[i]] for i in range(len(portfolio))]

Answer 4

简单你可以试试这个

a=list(zip(portfolio, index))