存储NumPy行和列标题

Question

我有一个numpy二维numpy数组，它包含多个股票的每日股价。例如

daily_prices = np.array([  
    [4,3,3,1],  
    [5,4,3,6],  
    [6,3,2,7],  
    [3,9,7,4],  
    [8,4,6,3],  
    [8,3,3,9]])  

其中每一行是不同的日期，每一列是不同的库存。

我希望能够存储在数组中（或更合适的东西），股票的名称（如'MSFT'，'CSCO'，'GOOG'，'F'）和日期下降。

换句话说，我想像在电子表格中那样命名行和列。

有没有什么新的方法可以做到这一点？

Answer 1

使用structured array：

import numpy as np

daily_prices = np.array(
    [
        (4,3,3,1),
        (5,4,3,6),
        (6,3,2,7),
        (3,9,7,4),
        (8,4,6,3),
        (8,3,3,9)],
    dtype=[('MSFT','float'),('CSCO','float'),('GOOG','float'),('F','float') ]
    )

这允许您访问如下列：

print(daily_prices['MSFT'])
# [ 4.  5.  6.  3.  8.  8.]

像这样的行：

print(daily_prices[2])
# (6.0, 3.0, 2.0, 7.0)