如何从numpy数组中提取子集

2024-09-29 13:30:02 发布

您现在位置:Python中文网/ 问答频道 /正文

让我们将以下列表命名为yr = [[90], [91, 92, 93, 94, 95]],将下面的单个元素列表命名为nm = ['abc', 'abd', 'fgm'],下面是一些numpy数组形式的数据,称为

d  = [ [95, 'abd', 'S', 432235],
       [94, 'abd', 'S', 432231],
       [91, 'abc', 'S', 343433],
       [92, 'abc', 'S', 342433],
       [94, 'abc', 'S', 344433],
       [90, 'abc', 'Z', 343234] ]

我要做的是遍历nm的每个元素以及yr的list元素,并提取相应的子集。例如,获取b的所有行,其中第二列等于'abs',第一列等于[90,91,92,93,94,95]中的任何一列。我怎么能重复这样做呢,也就是说,对于table nm的每个元素和yr的每个list元素?在

编辑:

这是我希望返回的输出:

^{pr2}$

Tags: 数据numpy元素列表abs数组命名子集
1条回答
网友
1楼 · 发布于 2024-09-29 13:30:02

我不明白空行,因为你应该得到18个元素(6个来自yr*3来自nm)。在

然后:

yr = [[90], [91, 92, 93, 94, 95]] 
nm = ['abc', 'abd', 'fgm'] 
d  = [[95, 'abd', 'S', 432235],
      [90, 'abc', 'S', 343433],
      [90, 'abc', 'Z', 343234]]


yr = [y for y in yr]    
for y in yr:
   for n in nm:
        for dd in d:
            print dd if dd[0] in y and dd[1] in n else '[[]]'

给出:

^{pr2}$

相关问题 更多 >