将第一行用作列名？Pandas read_html

Question

我有一个简单的一行脚本：

from pandas import read_html

print read_html('http://money.cnn.com/data/hotstocks/', flavor = 'bs4')

这个方法能够正常工作，但是列名丢失了，它们被识别为1、2、3。有没有一种简单的方法告诉pandas使用第一行作为列名呢？我知道我可以将名称存储为列表并设置它们，然后跳过第一行，但我想知道是否有更简单/更好的方法。

目前它打印出：

                           0       1       2         3
0                    Company   Price  Change  % Change
1             AAPL Apple Inc  115.31   +6.17    +5.65%
2   BAC Bank of America Corp   15.20   -0.43    -2.75%
3            YHOO Yahoo! Inc   46.46   -1.53    -3.19%
4        MSFT Microsoft Corp   41.19   -1.47    -3.45%
5            FB Facebook Inc   76.24   +0.46    +0.61%
6     GE General Electric Co   23.84   -0.54    -2.21%
7                 T AT&T Inc   32.68   -0.13    -0.40%
8            F Ford Motor Co   14.46   -0.24    -1.63%
9            INTC Intel Corp   33.78   -0.41    -1.20%
10    CSCO Cisco Systems Inc   26.80   -0.09    -0.35%

- nicholas.reichel

Answer 1

'read_html' 带有一个 header 参数。你可以传递一个行索引：

read_html('http://money.cnn.com/data/hotstocks/', header =0, flavor = 'bs4')

值得注意的是文档中的这个警告：

例如，如果在传递header=0参数时列名转换为NaN，则可能需要手动分配列名

http://pandas.pydata.org/pandas-docs/stable/generated/pandas.io.html.read_html.html

- JAB