过滤多维数组

Question

我有一个数组（列表），它是NxK。但是，我希望“过滤”是在根据第4列和第6列中的值输入一些约束之后。这是我到目前为止的代码。

INDEX

它有效，但不是那么整洁。只是想知道是否有人有更好的解决方案。

Answer 1

简化最重复的部分：

if k==0:
    for x in xrange(1,8):
        lists[i,x] = filter[0,x]
    k = 1
else:
    for x in xrange(1,8):
        lists[i,x] = add[0,x]
    filter = np.append(filter, add, axis=0)

您还可以将嵌套的if合并为一个，其中4个条件与and组合。

我也相信（没有看到lists是如何定义的，我不确定）你可以用

替换外循环

for row in lists:

然后使用row[x]代替lists[i,x]