python – 为2D数组实现numpy in1d？

发布时间：2020-12-16 21:50:48 所属栏目：Python 来源：网络整理

导读：我有一个2D numpy数组S代表一个状态空间,有80000000行(作为状态)和5列(作为状态变量). 我用S初始化K0,并且在每次迭代中,我对Ki中的所有状态应用状态转移函数f(x),并删除其f(x)不在Ki中的状态,得到Ki 1.直到它收敛,即Ki 1 = Ki. 这样做需要很长时间： K = Sto

我有一个2D numpy数组S代表一个状态空间,有80000000行(作为状态)和5列(作为状态变量).

我用S初始化K0,并且在每次迭代中,我对Ki中的所有状态应用状态转移函数f(x),并删除其f(x)不在Ki中的状态,得到Ki 1.直到它收敛,即Ki 1 = Ki.

这样做需要很长时间：

K = S
to_delete = [0]
While to_delete:
    to_delete = []
    for i in xrange(len(K)):
        if not f(i) in K:
        to_delete.append(K(i))
    K = delete(K,to_delete,0)

所以我想做一个矢量化的实现：

在列中切片K,应用f并再次连接它们,从而以某种方式获得f(K).

现在的问题是如何获得一个长度为len(K)的数组,比如Sel,其中每一行Sel [i]确定f(K [i])是否在K中.正好像in1d函数一样.

那么制作起来会很简单

K=K[Sel]]

最佳答案

您的问题很难理解,因为它包含无关的信息并包含拼写错误.如果我理解正确,你只需要一种有效的方法对2D数组的行执行设置操作(在这种情况下是K和f(K)的行的交集).

如果创建structured array视图,则可以使用numpy.in1d执行此操作.

码：

如果这是K：

In [50]: k
Out[50]:
array([[6,6],[3,7],[7,5],3],[1,8],[6,1],0]])

这是f(K)(对于这个例子,我从第一个col中减去1并在第二个col中加1)：

In [51]: k2
Out[51]:
array([[5,[2,4],[0,9],[5,2],1]])

然后你可以通过做这样的事情找到在f(K)中找到的K中的所有行：

In [55]: k[np.in1d(k.view(dtype='i,i').reshape(k.shape[0]),k2.view(dtype='i,i').
reshape(k2.shape[0]))]
Out[55]: array([[6,6]])

查看和重塑创建平面结构化视图,以便每行显示为in1d的单个元素. in1d创建匹配项的k的布尔索引,用于表示索引k并返回已过滤的数组.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!