Pandas：如何索引行？

Question

我有这样一个数据集：

stock_symbol    stock_date      stock_price_open     stock_price_high   
QRR             8/2/10  2.37    2.42                 2.29
QTM             5/2/10  2.38    2.5                  2.34
QXM             4/2/10  2.57    2.64                 2.39

我读了纽约证券交易所的数据： df = pandas.read_csv('NYSE.csv', index_col = 0, parse_dates=True)

当我跑的时候： df[['QRR','QTM','QXM']]

我有个错误： KeyError: "['QRR' 'QTM' 'QXM'] not in index"

'QRR'、'QTM'、'QXM'是stock\u symbol列的值。谁能告诉我怎么索引它们吗？你知道吗

Answer 1

必须使用^{}：

In [11]: df.loc[['QRR','QTM','QXM']]
Out[11]:
                stock_date  stock_price_open  stock_price_high
stock_symbol
QRR           8/2/10  2.37              2.42              2.29
QTM           5/2/10  2.38              2.50              2.34
QXM           4/2/10  2.57              2.64              2.39

---

如果没有loc，df[['QRR','QTM','QXM']]，pandas将尝试选择那些列（这些列不存在，因此出现“not in index”消息）：

In [21]: df[["stock_price_open", "stock_price_high"]]
Out[21]:
              stock_price_open  stock_price_high
stock_symbol
QRR                       2.42              2.29
QTM                       2.50              2.34
QXM                       2.64              2.39

查看文档的selecting data部分。你知道吗