Question

我有两个列表

kuid -> ['LOdRTFfn', 'Lckq4LkU', 'LcsYHodm']

NN是一个元组列表，其中第一个元素是kuid索引列表，另一个元素是值数组

NN -> [([0, 1, 2], [0.0, 1.2, 1.4]), ([1, 0, 2], [0.0, 1.4, 1.4]), ([2, 0, 1], [0.0, 1.1, 1.4])]

我只想保留值小于1.3的索引，例如：

[([0, 1], [0.0, 1.2]), ([1], [0.0]), ([2, 0], [0.0, 1.1])]

然后获取正确的kuid值并构建一个数据帧：

   kuid      la                    lametric                 
0  LOdRTFfn  [LOdRTFfn, Lckq4LkU]  [0.0, 1.2] 
1  Lckq4LkU  [Lckq4LkU]            [0.0]
2  LcsYHodm  [LcsYHodm, LOdRTFfn]  [0.0, 1.1]

是否可以使用理解列表（或其他快速解决方案）来避免循环？阵列可能非常大......

谢谢

Answer 1

这是单程

In [1789]: df = pd.DataFrame(dict(kuid=kuid, NN=[np.array(x) for x in NN]))

In [1790]: df['lametric'] = df.NN.apply(lambda x: x[1][x[1] < 1.3])

In [1791]: df['la'] = df.NN.apply(lambda x: [kuid[int(i)] for i in x[0][x[1] < 1.3]])

In [1792]: df
Out[1792]:
                                   NN      kuid    lametric  \
0  [[0.0, 1.0, 2.0], [0.0, 1.2, 1.4]]  LOdRTFfn  [0.0, 1.2]
1  [[1.0, 0.0, 2.0], [0.0, 1.4, 1.4]]  Lckq4LkU       [0.0]
2  [[2.0, 0.0, 1.0], [0.0, 1.1, 1.4]]  LcsYHodm  [0.0, 1.1]

                     la
0  [LOdRTFfn, Lckq4LkU]
1            [Lckq4LkU]
2  [LcsYHodm, LOdRTFfn]

从已过滤的元组列表构建数据框

1 个答案: