重塑numpy中的矩阵

Question

我目前正在尝试将三角矩阵重塑成这样的不同形式

              S&P 500   Russel 2000     Nasdaq Composite
S&P 500             1             0     0
Russel 2000         4             5     0
Nasdaq Composite    7             8     9

    0                           1                  
0   S&P 500                     S&P 500      
1   Russel 2000                 S&P 500
2   Nasdaq Composite            S&P 500
3   S&P 500                     Russel 2000
4   Russel 2000                 Russel 2000
...

也就是说，我正在尝试为每个索引创建重复的列，反之亦然。

我使用了一种称为J的面向数组的语言，但是对于python numpy来说，我还是一个新手。 你们能帮我解决这个问题吗？

在此先感谢=]

Answer 1

首先，这看起来是熊猫更容易解决的问题。

我已经进行了设置，可能会有所帮助，但是如果您真的不喜欢使用Numpy，那么它将不是解决问题的适当方法。

首先，我为您的numpy数组设置了类似的DataFrame对象：

$ import pandas as pd
$ df = pd.DataFrame(columns=['a','b','c'], index=['a','b','c'])
    a    b    c
a  NaN  NaN  NaN
b  NaN  NaN  NaN
c  NaN  NaN  NaN

然后，尽管我没有使用它们，但仍用您要求的值填充了它。

$ df['a'] = [1, 4, 7]
$ df['b'] = [0, 5, 8]
$ df['c'] = [0, 0, 9]
   a  b  c
a  1  0  0
b  4  5  0
c  7  8  9

最后，我只是简单地获取了索引和列，并使用了itertools中的乘积函数将每个组合添加到元组列表中。

$ from itertools import product
$ p = product(df.columns.values, df.index.values)
$ new_df = pd.DataFrame([i for i in p])
   0  1
0  a  a
1  a  b
2  a  c
3  b  a
4  b  b
5  b  c
6  c  a
7  c  b
8  c  c

我会看一下Pandas文档。 它们具有许多功能，可以按您感兴趣的方式操作数据框。