我有一个这样的dstack:
import numpy as np
a = np.array((1,2,6))
b = np.array((2,3,4))
c = np.array((8,3,0))
stack = np.dstack((a,b,c))
print(stack)
#[[[1 2 8]
#[2 3 3]
#[6 4 0]]]
我想过滤掉元素2小于1的列表。
大概是这样的
new_list = []
for i in stack:
for d in i[:,2]:
if d>=1:
new_list.append(d)
print(new_list) # [8,3]
这样做只添加了2个元素,但我希望有所有的行,如下所示:
#[[[1 2 8]
#[2 3 3]]]
如果I append(i)
,结果也不是期望的结果。
1条答案
按热度按时间ql3eal8s1#
你不需要一个循环,你可以用切片来完成
产出
如果您需要它作为
你可以把结果