如何存储NumPy行和列标题?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (1)
  • 关注 (0)
  • 查看 (26)

我有一个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’)和日期下降。

我想像在电子表格中那样命名行和列。有什么办法可以做到吗?

提问于
用户回答回答于

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)

扫码关注云+社区