python - 将两个列表转换为矩阵

Question

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

要绘制投资组合与市场指数的表现，我需要一个类似以下格式的数据结构:

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

但我将数据作为两个单独的一维数组:

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