Question

我是使用numpy类的新手，我在处理数组内容方面遇到了问题。这是代码：

# finance equation to apply to each element of array
for row in cum_ret:
   for col in row:
       if sum(row)!=0:
           row[col] = prev_row[col]*(1+row[col])
       else:
           row[col] = 1

       cum_ret[row][col] = row[col]
   prev_row = row
# see changed contents
for row in cum_ret:
print row

现在我收到一个错误，说使用的数组索引必须是整数或布尔类型。我得到它，因为'row'值也是一个数组，因此它不能索引数组对象。那么这样做的正确语法是什么，或者是我应该使用的方法呢？

提前感谢

cum_ret数组是float64s的2d ndarray，是我想要修改的数组。以下是输出的简短片段：

[[ 0.          0.          0.          0.          0.        ]
 [ 0.00046187  0.00836672  0.00020435 -0.00048292  0.00342209]
 [-0.07633505 -0.00514199 -0.04133778 -0.02450642 -0.01865075]
 ..., 
 [ 0.01229435  0.00175341  0.00709808  0.00213371  0.0061171 ]
 [-0.0118614  -0.00994933 -0.00557095 -0.00141945 -0.00347423]
 [ 0.01214725 -0.00502466  0.00537611 -0.00035537 -0.00101685]]

现在它正在发生：

Traceback (most recent call last):
File "qstk1.py", line 37, in <module>
   cum_ret[row][col] = row[col]

IndexError: arrays used as indices must be of integer (or boolean) type

Answer 1

如果cum_ret是数组，您可以使用numpy.sum(cum_ret, axis=1)获取行和，并numpy.sum(cum_ret, axis=1) != 0立即在整个数组上生成测试。然后，您可以使用numpy.select()来应用您的条件。

如果你在循环的开头加上一个print row语句，你会注意到它不是整数，而是一个numpy数组......这会导致你的错误。顺便说一下，你不需要任何这样的循环来执行这个操作。

另一个考虑因素：第一行会发生什么？什么被认为是前一行？

编辑：

阅读完评论后，我想你想要这样的事情：

import numpy
cum_ret = numpy.array([[0,0,0,0],[-0.234,-0.365,-0.634,-0.453], [-0.334,-0.465,-0.534,-0.653],[-0.134,-0.265,-0.334,-0.453]])
b = cum_ret + 1
c = numpy.cumprod(b, axis=0)

不需要循环，也无需检查零行的条件。

计算Numpy中的累积回报，数组索引的问题

1 个答案: