python - 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

我读了一个 NYSE 数据集:df = pandas.read_csv('NYSE.csv', index_col = 0, parse_dates=True)

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

我遇到了这个错误:KeyError: “['QRR' 'QTM' 'QXM'] 不在索引中”

'QRR'、'QTM'、'QXM' 是 stock_symbol 列的值。谁能告诉我如何将它们编入索引？

Answer 1

你必须使用 loc :

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 会尝试选择那些不存在的列(因此“不在索引中” "消息):

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文档部分。