Python中嵌套列表的算术运算

时间:2012-08-09 18:16:39

标签: python numpy python-2.7 nested-lists

我试图减去嵌套列表中的值(来自雅虎财务的历史股票价格数据列表),我遇到了问题。我正在尝试简单的减法(即高 - 低),但我无法实现这一点。我可能遗漏了一些关于列表性质的基本信息,但我很难过。

我正在使用的嵌套列表的一个示例:

[['2012-07-31', '16.00', '16.06', '15.81', '15.84', '13753800', '15.8'],
 ['2012-07-30', '16.15', '16.15', '15.90', '15.98', '10187600', '15.9'],
 ['2012-07-27', '15.88', '16.17', '15.84', '16.11', '14220800', '16.1'],
 ['2012-07-26', '15.69', '15.88', '15.62', '15.80', '11033300', '15.8'],
 ['2012-07-25', '15.52', '15.64', '15.40', '15.50', '15092000', '15.5'],
 ['2012-07-24', '15.74', '15.76', '15.23', '15.43', '19733400', '15.4'],
 ['2012-07-23', '15.70', '15.81', '15.59', '15.76', '14825800', '15.7'],
 ['2012-07-20', '15.75', '15.94', '15.68', '15.92', '16919700', '15.9'],
 ['2012-07-19', '15.71', '15.86', '15.64', '15.73', '15985300', '15.7'],
 ...]

我想从第三个“列”中减去第4个“列”并用结果填充另一个列表(命令IS很重要。)实现此目的的最佳方法是什么?

4 个答案:

答案 0 :(得分:3)

您可以使用列表理解:

from decimal import Decimal
result = [(row[0], Decimal(row[2]) - Decimal(row[3])) for row in data]

答案 1 :(得分:1)

您的第一个问题是您的列表包含字符串而不是数字。如果您想减去将它们转换为数字所需的数字(例如,在您创建列表时使用float)。

似乎您希望在组件中添加和减去数字,将数据视为包含行和列的表。为此,您应该使用numpy,这是为此设计的库。您可能还想查看pandas,这是一个基于numpy构建的库,它提供了按行和列切片和切割数据的强大功能。 (如果您正在从文件中读取数据,这些库还提供了一些可以让您阅读的工具,例如CSV文件,并将数据作为数字和/或日期而不是字符串导入。)

答案 2 :(得分:1)

执行此操作的最佳方法是使用numpy,python从未被设计为处理大量数据,numpy是因为其子程序的大量使用其他语言编译为本机二进制文件,它可以使用加速线性代数库来真正加速计算。

这是一个简单的例子:

>>> import numpy
>>> values = numpy.random.rand(5, 5) # 5 by 5 matrix with random values
>>> values[:, 3] - values[:, 2] # numpy is 0 index, so the fourth column is 3 and the third is 2

答案 3 :(得分:1)

在原生Python中,如果你想保留嵌套列表(称之为'table';其中的每个列表都是'row'),那么创建差异列表的简洁,惯用方法是:

differences = [float(row[3]) - float(row[4]) for row in table]

以便differences[i] == table[i][3] - table[i][4]

如果表中的数字数据将被其他代码使用,您可能希望将字符串转换为表中的浮点数:

table = [[r[0], float(r[1]), float(r[2]),
         float(r[3]), float(r[4]), r[5], float(r[6])] for r in table]

这样差异表就可以由

创建
differences = [r[3] - r[4] for r in table]