我想用一个索引过滤我的数组2D,然后只用过滤器中的值来调整这个数组。这差不多是什么tf.sequence_掩码可以,但我需要这个在纽比或其他光图书馆。在
谢谢!在
警察局:
这是一个例子:
array_2d = [[0,1,2,3,4,5],[8,9,10,11,12,0],[21,22,21,0,0,0]] # this is a numpy array
array_len = [6,5,3]
expected_output = [0,1,2,3,4,5,8,9,10,11,12,21,22,21]
Tags:
这是一种使用布尔蒙版并将其应用于展平的
array_2d
的方法输出
^{pr2}$下面是一个
vectorized
解决方案,使用布尔掩码索引array_2d
:详细信息
创建掩码时将^{} 放在与} 上,并执行行比较以查看哪些元素大于
array_2d
形状相同的^{array_len
。在因此,第一步是创建以下
^{pr2}$ndarray
:并与
array_len
执行行比较:然后您只需使用以下内容筛选数组:
相关问题 更多 >
编程相关推荐