使用 python 在 .csv 文件中添加标头

Question

我当前的代码能够从 Google Finance 获取历史数据，但缺少第一行来标记每一列。 我想要如下的输出

库存、日期、时间、开盘价、最高价、最低价、收盘价、成交量 CCK,2015-12-30,00:00:00,1.05,1.05,0.99,1.00,157800 CCK,2015-12-31,00:00:00,0.98,1.03,0.98,1.02,55300

 stocklist=['CCK','MSFT','AA','AAPL']
        stocklen=len(stocklist)

for x in range(1,stocklen,1):
    q = GoogleQuote(stocklist[x],'2015-12-21')
    header = ['Stock','Date','Time','Open','High','Low','Close']
    print header                                             
    print q                                                   
    q.write_csv(stocklist[x]+".csv")
    with open(stocklist[x]+".csv",'a') as f:
        w = csv.writer(f)
        w.writerow(['Stock','Date','Time','Open','High','Low','Close'])

但是，输出在最后一行

CCK,2015-12-30,00:00:00,1.05,1.05,0.99,1.00,157800 CCK,2015-12-31,00:00:00,0.98,1.03,0.98,1.02,55300 股票，日期，时间，开盘，高，低，收盘，成交量

Answer 1

GoogleQuote 来自哪里？是来自quotes.py吗？？？

如果是这样，那么您的代码将已经生成带有标题行的 CSV 文件，唯一的区别是第一列标记为 Symbol 而不是 Stock。如果您反对该标签，您可以像这样更改它：

import csv
import datetime
from itertools import repeat

header = ['Stock', 'Date', 'Time', 'Open', 'High', 'Low', 'Close', 'Volume']
stocklist=['CCK','MSFT','AA','AAPL']
start = datetime.datetime(2015, 12, 21)
#end = datetime.datetime(2015, 12, 31)
for stock in stocklist:
    q = GoogleQuote(stock, start.strftime('%F'))
    data = zip(repeat(q.symbol), q.date, q.time, q.open_, q.high, q.low, q.close, q.volume)
    with open('{}.csv'.format(stock), 'w') as f:
        w = csv.writer(f)
        w.writerow(header)
        w.writerows(data)

---

你可以使用pandas:

import datetime
import pandas.io.data as web

stocklist=['CCK','MSFT','AA','AAPL']
start = datetime.datetime(2015, 12, 21)
end = datetime.datetime(2015, 12, 31)
for stock in stocklist:
    data = web.DataReader(stock, 'google', start, end)
    data.to_csv('{}.csv'.format(stock))

这将为每只股票在包含日期的范围内创建单独的 CSV 文件，并带有标题行：

日期、开盘价、最高价、最低价、收盘价、成交量

pandas 和你的代码在输出上的区别是：

• 库存不作为 CSV 的第一列显示。由于每个文件都被命名 根据库存，文件本身不需要它。
• 时间列不存在。报价是针对日期的，因此时间可能不相关。

如果您确实必须有股票和时间列，可以将它们添加到数据框中。

for stock in stocklist:
    data = web.DataReader(stock, 'google', start, end)
    data.insert(0, 'Time', [ts.time() for ts in data.index])
    data.reset_index(inplace=True)    # make Date a proper column
    data.insert(0, 'Stock', stock)
    data.to_csv('{}.csv'.format(stock), index=False)

请注意，日期列是时间戳，因此可能不需要添加时间列。相反，您可以省略添加时间列并使用以下格式设置日期：

    data.to_csv('{}.csv'.format(stock), index=False, date_format='%F %T')