2

我对 Python 很陌生，想构建一个黑匣子股票交易程序，它可以找到股票回报率之间的各种相关性，并给我一个响应，例如买入、卖出、持有等。我发现了一个简洁易用的 Python 模块用于检索名为 ystockquote 的股票数据，该数据从 Yahoo! 金融。该模块可以在http://www.goldb.org/ystockquote.html找到。

它的功能之一是以 形式输出股票的历史价格['Date', 'Open', 'High', 'Low', 'Close', 'Volume', 'Adj Clos']。我可以给它一个日期范围来执行此操作，它给我一个嵌套列表，其中包含一个列表，其中包含每天的上述信息。

我的问题是如何将这些单独的数据点（日期、开盘价、高点、低点等）组织成一个结构，以便稍后在我的脚本中调用并进行排序。我需要这个过程易于自动化。我会发现哪些类型的算法或数据结构有用？

4

2 回答 2

5

您可能正在寻找字典结构而不是列表：

>>> prices = dict()
>>> prices['2011-01-02'] = {'Open':20.00, 'High':30.00, 'Low':10.00, 'Close':21.00, 'Volume':14.00, 'Adj Clos':120}
>>> prices['2010-11-09'] = {'Open':22.00, 'High':50.00, 'Low':20.00, 'Close':42.00, 'Volume':10.00, 'Adj Clos':666}
>>> prices
{'2011-01-02': {'Volume': 14.0, 'Adj Clos': 120, 'High': 30.0, 'Low': 10.0, 'Close': 21.0, 'Open': 20.0}, '2010-11-09': {'Volume': 10.0, 'Adj Clos': 666, 'High': 50.0, 'Low': 20.0, 'Close': 42.0, 'Open': 22.0}}

在这里，我在主“价格”字典的每个条目中嵌套了一个字典。字典的第一级以日期为键，并映射到包含该日期价格信息的字典。

>>> prices['2011-01-02']
{'Volume': 14.0, 'Adj Clos': 120, 'High': 30.0, 'Low': 10.0, 'Close': 21.0, 'Open': 20.0}

字典的第二级使用属性名称作为键，并映射到属性值本身。

>>> prices['2010-11-09']['Open']
22.0
>>> prices['2010-11-09']['Close']
42.0

看来，对于get_historical_prices您所指的功能，每天都作为表单的条目输出[Date, Open, High, Low, Close, Volume, Adj_Clos]。如果你想为这些条目的列表构建一个字典，你需要做三件事：

首先，您需要索引每个条目以将Date与其他元素分开，因为这是您将用来索引 dict 的第一个维度的内容。您可以使用 获取第一个元素，使用 获取entry[0]剩余元素entry[1:]。

>>> entry = ['2011-01-02', 20.00, 30.00, 10.00, 21.00, 14.00, 120]
>>> date = entry[0]
>>> date
'2011-01-02'
>>> values = entry[1:]
>>> values
[20.0, 30.0, 10.0, 21.0, 14.0, 120]

其次，由于您希望将其他每个元素与特定键相关联，因此您应该按照与数据元素提供给您的顺序相同的顺序列出这些键。使用该zip()函数，您可以组合两个列表p和q，从每个列表中获取第 i 个元素并生成zip(p,q)[i] == (p[i], q[i])。通过这种方式，您可以创建一个 (key, value) 对列表，您可以将其传递给字典构造函数：

>>> keys = ['Open', 'High', 'Low', 'Close', 'Volume', 'Adj Clos']
>>> pairs = zip(keys, entry[1:])
>>> pairs
[('Open', 20.0), ('High', 30.0), ('Low', 10.0), ('Close', 21.0), ('Volume', 14.0), ('Adj Clos', 120)]

最后，您要构建字典，并将其索引到整个历史中的适当日期：

>>> stockdict = dict(pairs)
>>> stockdict
{'Volume': 14.0, 'Adj Clos': 120, 'High': 30.0, 'Low': 10.0, 'Close': 21.0, 'Open': 20.0}
>>> histodict = dict()
>>> histodict[date] = stockdict

您可以通过两种方式遍历嵌套history列表来构造字典，第一种是使用传统for循环：

keys = ['Open', 'High', 'Low', 'Close', 'Volume', 'Adj Clos']
histodict = dict()
for item in history:
    date = item[0]
    values = item[1:]
    histodict[date] = dict(zip(keys, values))

或者，如果您想使用更高级的 Python 技术，请尝试使用嵌套的 dict生成器语句：

keys = ['Open', 'High', 'Low', 'Close', 'Volume', 'Adj Clos']
histodict = dict((item[0], dict(zip(keys, item[1:]))) for item in history)

如果您是编程新手，最后一个会很麻烦，但我鼓励您阅读该链接；请记住，使用 Python 编程时，Google 是您的朋友。我希望我在这里给了你足够的关键词和想法来开始学习，剩下的就交给你了。

于 2011-06-16T21:57:17.187 回答
1

给定一个相等长度的列表，很容易按任何“列”排序：

>>> l = [[1, 2, 3, 4, 5], [5, 4, 3, 2, 1], [0, 0, 0, 0, 0], [6, 6, 6, 6, 6]]
>>> l.sort(key=lambda l: l[1])
>>> l
[[0, 0, 0, 0, 0], [1, 2, 3, 4, 5], [5, 4, 3, 2, 1], [6, 6, 6, 6, 6]]
>>> l.sort(key=lambda l: l[4])
>>> l
[[0, 0, 0, 0, 0], [5, 4, 3, 2, 1], [1, 2, 3, 4, 5], [6, 6, 6, 6, 6]]

关键字参数接受一个函数，给定列表中的key一个项目，返回一个用作排序键的值。

但是如果你想做更多有趣的事情，你可能最好使用数据库。为方便起见（对我而言），sqlite3文档使用了一个股票表作为示例，我已经对其进行了适当的修改和修改：

import sqlite3
conn = sqlite3.connect('/tmp/example')   # use ':memory:' for an in-memory db
c = conn.cursor()

# Create table
c.execute('''create table stocks
(date text, trans text, symbol text,
 qty real, price real)''')

# Insert a row of data
c.execute("""insert into stocks
          values ('2006-01-05','BUY','RHAT',100,35.14)""")

# Save (commit) the changes
conn.commit()

# Insert another row of data
c.execute("""insert into stocks
          values ('2006-01-07','SELL','RHAT',100,2.11)""")

# Select rows of data from table in an order
rows_by_date = c.execute("""select * from stocks order by date""")
for row in rows_by_date:
    print row

# In a different order
rows_by_price = c.execute("""select * from stocks order by price""")
for row in rows_by_price:
    print row
于 2011-06-16T21:59:38.087 回答